fffiloni commited on
Commit
feb2a2f
·
1 Parent(s): 116c46b

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -7
Dockerfile CHANGED
@@ -20,13 +20,8 @@ RUN git clone -b dev https://github.com/camenduru/DiffBIR $HOME/app
20
  # Install Python dependencies
21
  RUN pip install -q einops pytorch_lightning gradio omegaconf xformers==0.0.20 transformers lpips
22
 
23
- # Install Hugging Face Transformers library
24
- RUN pip install transformers
25
-
26
- # Install Hugging Face Datasets library (if needed)
27
- # RUN pip install datasets
28
-
29
- # Install other dependencies as required
30
 
31
  # Download checkpoint files using aria2
32
  RUN apt-get update && apt-get install -y aria2
 
20
  # Install Python dependencies
21
  RUN pip install -q einops pytorch_lightning gradio omegaconf xformers==0.0.20 transformers lpips
22
 
23
+ # Install open_clip from GitHub
24
+ RUN pip install -q git+https://github.com/mlfoundations/open_clip@v2.20.0
 
 
 
 
 
25
 
26
  # Download checkpoint files using aria2
27
  RUN apt-get update && apt-get install -y aria2