Spaces:

TRI-ML
/

vlm-demo

Paused

mattb512 commited on Apr 1, 2024

Commit

98873da

1 Parent(s): 3cfff40

add yamlinclude package

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -6,7 +6,7 @@ RUN apt-get -y update \
     && rm -rf /var/lib/apt/lists/*
 RUN python3 -m pip install --upgrade pip \
-    && pip install --no-cache-dir  torch torchvision torchaudio pyyaml draccus
 RUN useradd -m -u 1000 user

     && rm -rf /var/lib/apt/lists/*
 RUN python3 -m pip install --upgrade pip \
+    && pip install --no-cache-dir  torch torchvision torchaudio pyyaml draccus pyyaml-include
 RUN useradd -m -u 1000 user