fffiloni commited on
Commit
37a3a7f
·
1 Parent(s): 180ff73

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -22,7 +22,7 @@ WORKDIR $HOME/app
22
  RUN git clone -b dev https://github.com/camenduru/DiffBIR $HOME/app
23
 
24
  # Install Python dependencies
25
- RUN pip install -q einops pytorch_lightning gradio omegaconf xformers==0.0.16 transformers lpips opencv-python
26
 
27
  # Install open_clip from GitHub
28
  RUN pip install -q git+https://github.com/mlfoundations/open_clip@v2.20.0
 
22
  RUN git clone -b dev https://github.com/camenduru/DiffBIR $HOME/app
23
 
24
  # Install Python dependencies
25
+ RUN pip install -q xformers==0.0.16 pytorch_lightning==1.4.2 einops open-clip-torch omegaconf torchmetrics==0.6.0 triton opencv-python-headless scipy matplotlib lpips gradio chardet transformers facexlib
26
 
27
  # Install open_clip from GitHub
28
  RUN pip install -q git+https://github.com/mlfoundations/open_clip@v2.20.0