fffiloni commited on
Commit
bbed12c
1 Parent(s): d07de3e

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -25,7 +25,7 @@ ENV HOME=/home/user \
25
  WORKDIR $HOME/app
26
 
27
  # Clone your repository or add your code to the container
28
- RUN git clone -b dev https://github.com/fffiloni/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
 
25
  WORKDIR $HOME/app
26
 
27
  # Clone your repository or add your code to the container
28
+ RUN git clone -b main https://github.com/fffiloni/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