Spaces:
Paused
Paused
Update Dockerfile
Browse files- 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
|
29 |
|
30 |
# Install open_clip from GitHub
|
31 |
-
RUN
|
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
|