diff --git a/Dockerfile b/Dockerfile index 3c38db3..ab46a52 100644 --- 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 .