Spaces:
Running
on
Zero
Running
on
Zero
Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
@@ -26,7 +26,7 @@ WORKDIR $HOME/app
|
|
26 |
|
27 |
# Clone your repository or add your code to the container
|
28 |
RUN git clone -b dev https://github.com/camenduru/PASD $HOME/app
|
29 |
-
|
30 |
RUN pip install -q gradio gradio_imageslider diffusers==0.21.4 accelerate basicsr ultralytics salesforce-lavis webdataset pytorch_lightning
|
31 |
RUN pip install -q https://download.pytorch.org/whl/cu121/xformers-0.0.22.post7-cp310-cp310-manylinux2014_x86_64.whl
|
32 |
|
|
|
26 |
|
27 |
# Clone your repository or add your code to the container
|
28 |
RUN git clone -b dev https://github.com/camenduru/PASD $HOME/app
|
29 |
+
RUN pip install torchvision==0.15.2 torchaudio==2.0.2
|
30 |
RUN pip install -q gradio gradio_imageslider diffusers==0.21.4 accelerate basicsr ultralytics salesforce-lavis webdataset pytorch_lightning
|
31 |
RUN pip install -q https://download.pytorch.org/whl/cu121/xformers-0.0.22.post7-cp310-cp310-manylinux2014_x86_64.whl
|
32 |
|