fffiloni commited on
Commit
f02dff4
1 Parent(s): ff50340

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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