Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -31,7 +31,7 @@ RUN git clone -b main https://github.com/fffiloni/PASD $HOME/app
|
|
| 31 |
# RUN pip install torchaudio --upgrade
|
| 32 |
# RUN pip install torchvision
|
| 33 |
RUN pip install -U torch torchaudio torchvision --no-cache-dir
|
| 34 |
-
RUN pip install -q spaces gradio==3.39.0 diffusers==0.21.4 accelerate basicsr ultralytics salesforce-lavis webdataset pytorch_lightning=2.1.
|
| 35 |
RUN pip install -q https://download.pytorch.org/whl/cu121/xformers-0.0.22.post7-cp310-cp310-manylinux2014_x86_64.whl
|
| 36 |
|
| 37 |
RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/akhaliq/RetinaFace-R50/resolve/main/RetinaFace-R50.pth -d $HOME/app/annotator/ckpts -o RetinaFace-R50.pth
|
|
|
|
| 31 |
# RUN pip install torchaudio --upgrade
|
| 32 |
# RUN pip install torchvision
|
| 33 |
RUN pip install -U torch torchaudio torchvision --no-cache-dir
|
| 34 |
+
RUN pip install -q spaces gradio==3.39.0 diffusers==0.21.4 accelerate basicsr ultralytics salesforce-lavis webdataset pytorch_lightning=2.1.4
|
| 35 |
RUN pip install -q https://download.pytorch.org/whl/cu121/xformers-0.0.22.post7-cp310-cp310-manylinux2014_x86_64.whl
|
| 36 |
|
| 37 |
RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/akhaliq/RetinaFace-R50/resolve/main/RetinaFace-R50.pth -d $HOME/app/annotator/ckpts -o RetinaFace-R50.pth
|