fffiloni commited on
Commit
c8e0d2e
1 Parent(s): 1da07cd

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -2
Dockerfile CHANGED
@@ -27,8 +27,9 @@ RUN pip install -q einops pytorch_lightning gradio omegaconf xformers==0.0.16 tr
27
  # Install open_clip from GitHub
28
  RUN pip install -q git+https://github.com/mlfoundations/[email protected]
29
 
30
- # Install aria2 for downloading files
31
- RUN apt-get update && apt-get install -y aria2
 
32
 
33
  # Download checkpoint files using aria2
34
  RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/lxq007/DiffBIR/resolve/main/general_full_v1.ckpt -d $HOME/app/models -o general_full_v1.ckpt
 
27
  # Install open_clip from GitHub
28
  RUN pip install -q git+https://github.com/mlfoundations/[email protected]
29
 
30
+ # Update package lists and install aria2 without sudo
31
+ RUN apt-get update && apt-get install -y --no-install-recommends aria2
32
+
33
 
34
  # Download checkpoint files using aria2
35
  RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/lxq007/DiffBIR/resolve/main/general_full_v1.ckpt -d $HOME/app/models -o general_full_v1.ckpt