Add open-clip-torch dependency and fix import in degradations.py

rainbowdashx · Feb 17, 2024 · 0323468 · 0323468
1 parent 01c6201
commit 0323468
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -76,6 +76,8 @@ RUN --mount=type=cache,target=/root/.cache/pip \
     pip install --upgrade -r /requirements.txt --no-cache-dir && \
     rm /requirements.txt
 
+RUN pip install open-clip-torch==2.20.0
+
 RUN sed -i 's/from torchvision.transforms.functional_tensor import rgb_to_grayscale/from torchvision.transforms.functional import rgb_to_grayscale/g' /usr/local/lib/python3.10/site-packages/basicsr/data/degradations.py
 
 ADD src .