cache fix

README.md

@@ -38,6 +38,8 @@ datasets:
 preload_from_hub:
 - Aleksandar/PartEdit-Bench
 - Aleksandar/PartEdit-extra
+- stabilityai/stable-diffusion-xl-base-1.0
+- madebyollin/sdxl-vae-fp16-fix
 thumbnail: >-
   https://cdn-uploads.huggingface.co/production/uploads/61127c19e0075d7cf22d6bc2/JV1dztJb76Sl_HGYPlc4O.jpeg
 ---

app.py

@@ -21,22 +21,22 @@ import pathlib
 
 # ---- Robust HF cache setup for Spaces (persistent or not) ----
 
-def _pick_hf_cache_base() -> str:
-    data = "/data"
-    if os.path.isdir(data) and os.access(data, os.W_OK):
-        return "/data/.huggingface"                 # persistent disk on Spaces
-    return "/tmp/hf_cache"                           # ephemeral but writable everywhere
-
-HF_BASE = os.environ.get("HF_HOME") or _pick_hf_cache_base()
-HF_BASE = str(pathlib.Path(HF_BASE).absolute())
-
-os.environ.setdefault("HF_HOME", HF_BASE)                           # shared root (recommended)
-os.environ.setdefault("HF_HUB_CACHE", f"{HF_BASE}/hub")             # hub cache (models/datasets/spaces)
-os.environ.setdefault("HF_DATASETS_CACHE", f"{HF_BASE}/datasets")   # datasets Arrow cache
-os.environ.setdefault("TRANSFORMERS_CACHE", f"{HF_BASE}/hub")       # transformers cache
-
-for k in ("HF_HUB_CACHE", "HF_DATASETS_CACHE", "TRANSFORMERS_CACHE"):
-    os.makedirs(os.environ[k], exist_ok=True)
+# def _pick_hf_cache_base() -> str:
+#     data = "/data"
+#     if os.path.isdir(data) and os.access(data, os.W_OK):
+#         return "/data/.huggingface"                 # persistent disk on Spaces
+#     return "/tmp/hf_cache"                           # ephemeral but writable everywhere
+
+# HF_BASE = os.environ.get("HF_HOME") or _pick_hf_cache_base()
+# HF_BASE = str(pathlib.Path(HF_BASE).absolute())
+
+# os.environ.setdefault("HF_HOME", HF_BASE)                           # shared root (recommended)
+# os.environ.setdefault("HF_HUB_CACHE", f"{HF_BASE}/hub")             # hub cache (models/datasets/spaces)
+# os.environ.setdefault("HF_DATASETS_CACHE", f"{HF_BASE}/datasets")   # datasets Arrow cache
+# os.environ.setdefault("TRANSFORMERS_CACHE", f"{HF_BASE}/hub")       # transformers cache
+
+# for k in ("HF_HUB_CACHE", "HF_DATASETS_CACHE", "TRANSFORMERS_CACHE"):
+#     os.makedirs(os.environ[k], exist_ok=True)
 # ---------------------------------------------------------------
 
 MAX_SEED = np.iinfo(np.int32).max
@@ -45,7 +45,7 @@ AVAILABLE_TOKENS = list(PART_TOKENS.keys())
 
 # Download examples directly from the huggingface PartEdit-Bench
 # Login using e.g. `huggingface-cli login` or `hf login` if needed.
-bench = load_dataset("Aleksandar/PartEdit-Bench", revision="v1.1", split="synth", cache_dir=os.environ["HF_DATASETS_CACHE"])
+bench = load_dataset("Aleksandar/PartEdit-Bench", revision="v1.1", split="synth", cache_dir="~/.cache/huggingface/hub")
 
 use_examples = None  # all with None
 logo = "assets/partedit.png"