From 7638fb4e5ee45f54899ae2da02c64c11127f2aa3 Mon Sep 17 00:00:00 2001
From: AbdBarho <ka70911@gmail.com>
Date: Sat, 24 Sep 2022 09:57:57 +0200
Subject: [PATCH] Fix CLIP model caching #88 (#95)

Refs #88
hacky solution but works for now
---
 services/AUTOMATIC1111/Dockerfile | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/services/AUTOMATIC1111/Dockerfile b/services/AUTOMATIC1111/Dockerfile
index 0ebc9aa..ce1f252 100644
--- a/services/AUTOMATIC1111/Dockerfile
+++ b/services/AUTOMATIC1111/Dockerfile
@@ -61,8 +61,12 @@ RUN pip install --prefer-binary -U --no-cache-dir opencv-python-headless
 ENV TRANSFORMERS_CACHE=/cache/transformers TORCH_HOME=/cache/torch CLI_ARGS=""
 
 COPY . /docker
-RUN chmod +x /docker/mount.sh && python3 /docker/info.py ${ROOT}/modules/ui.py
-
+RUN <<EOF
+chmod +x /docker/mount.sh && python3 /docker/info.py ${ROOT}/modules/ui.py
+# hackiest of hacks, change default cache dir of clip #88
+# https://github.com/openai/CLIP/blob/d50d76daa670286dd6cacf3bcd80b5e4823fc8e1/clip/clip.py#L94
+sed -i -- 's/download_root: str = None/download_root: str = "\/cache\/weights"/' /opt/conda/lib/python3.8/site-packages/clip/clip.py
+EOF
 
 WORKDIR ${WORKDIR}
 EXPOSE 7860