fffiloni commited on
Commit
7b96708
1 Parent(s): 71a1790

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -57,6 +57,7 @@ RUN pip install -r requirements_HF.txt
57
  RUN pip install gradio
58
 
59
  USER root
 
60
  RUN pip install cupy-cuda11x
61
  # Install basicsr (assuming setup.py is in basicsr directory)
62
  RUN python basicsr/setup.py develop
 
57
  RUN pip install gradio
58
 
59
  USER root
60
+ RUN pip install setuptools
61
  RUN pip install cupy-cuda11x
62
  # Install basicsr (assuming setup.py is in basicsr directory)
63
  RUN python basicsr/setup.py develop