fffiloni commited on
Commit
b69a5bc
·
1 Parent(s): 48eeb5a

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -25,10 +25,10 @@ RUN git clone -b dev https://github.com/camenduru/DiffBIR $HOME/app
25
  #RUN pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html
26
 
27
  # Install Python dependencies
28
- RUN pip install -q einops pytorch_lightning gradio omegaconf xformers transformers lpips opencv-python
29
 
30
  # Install open_clip from GitHub
31
- RUN pip install -q git+https://github.com/mlfoundations/open_clip@v2.20.0
32
 
33
  # Use sudo to install aria2 with elevated privileges
34
  USER root
 
25
  #RUN pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html
26
 
27
  # Install Python dependencies
28
+ RUN pip3 install -q einops pytorch_lightning gradio omegaconf xformers==0.0.20 transformers lpips opencv-python
29
 
30
  # Install open_clip from GitHub
31
+ RUN pip3 install -q git+https://github.com/mlfoundations/open_clip@v2.20.0
32
 
33
  # Use sudo to install aria2 with elevated privileges
34
  USER root