Update pipeline/pipeline.py

pipeline/pipeline.py

@@ -1,3 +1,35 @@
+import os
+from pathlib import Path
+
+# CACHE PATCH BLOCK: place FIRST in pipeline.py!
+HF_CACHE_DIR = Path("/tmp/hf_cache")
+HF_CACHE_DIR.mkdir(parents=True, exist_ok=True)
+os.environ.update({
+    "HF_HOME": str(HF_CACHE_DIR),
+    "HF_HUB_CACHE": str(HF_CACHE_DIR),
+    "DIFFUSERS_CACHE": str(HF_CACHE_DIR),
+    "TRANSFORMERS_CACHE": str(HF_CACHE_DIR),
+    "XDG_CACHE_HOME": str(HF_CACHE_DIR),
+    "HF_DATASETS_CACHE": str(HF_CACHE_DIR),
+    "HF_MODULES_CACHE": str(HF_CACHE_DIR),
+    "TMPDIR": str(HF_CACHE_DIR),
+    "CACHE_DIR": str(HF_CACHE_DIR),
+    "TORCH_HOME": str(HF_CACHE_DIR),
+    "HOME": str(HF_CACHE_DIR)
+})
+import os.path
+if not hasattr(os.path, "expanduser_original"):
+    os.path.expanduser_original = os.path.expanduser
+def safe_expanduser(path):
+    if (
+        path.startswith("~") or 
+        path.startswith("/.cache") or 
+        path.startswith("/root/.cache")
+    ):
+        return str(HF_CACHE_DIR)
+    return os.path.expanduser_original(path)
+os.path.expanduser = safe_expanduser
+
 import asyncio
 import logging
 import core.script_gen as script_gen