Prefetch models with the parallel Hugging Face downloader

krea2-start.py

@@ -79,6 +79,44 @@ if not has_krea:
     rc = subprocess.run(cmd + f" >>{LOG} 2>&1", shell=True).returncode
     log(f"comfyui upgrade exit {rc}")
 
+# The bootstrap's own downloader uses one connection, and from RunPod machines
+# Hugging Face transfers stall on it (seen at 120 MB, 2026-09-25). Prefetch the
+# three files with Hugging Face's parallel downloader into the exact paths the
+# bootstrap checks; it then finds them verified and skips its own download.
+PREFETCH = r'''
+import os, shutil, sys, time, urllib.parse
+from huggingface_hub import hf_hub_download
+url, dest = sys.argv[1], sys.argv[2]
+parts = urllib.parse.urlparse(url).path.strip("/").split("/")
+repo, rev, path = "/".join(parts[:2]), parts[3], urllib.parse.unquote("/".join(parts[4:]))
+t = time.time()
+local = hf_hub_download(repo_id=repo, filename=path, revision=rev, local_dir="/tmp/hfdl",
+                        token=os.environ.get("LOG_TOKEN") or None)
+os.makedirs(os.path.dirname(dest), exist_ok=True)
+shutil.move(local, dest)
+size = os.path.getsize(dest)
+print(f"prefetched {os.path.basename(dest)} {size/1e9:.2f} GB in {time.time()-t:.0f}s ({size/1e6/max(time.time()-t,1):.0f} MB/s)")
+'''
+rc = subprocess.run("python3 -m pip install -q --target /opt/hfdl 'huggingface_hub>=0.32,<1.0' hf_xet "
+                    f">>{LOG} 2>&1 || uv pip install -q --target /opt/hfdl 'huggingface_hub>=0.32,<1.0' hf_xet >>{LOG} 2>&1",
+                    shell=True).returncode
+log(f"hf downloader install exit {rc}")
+open("/opt/prefetch.py", "w").write(PREFETCH)
+env = dict(os.environ, PYTHONPATH="/opt/hfdl", HF_HUB_DISABLE_PROGRESS_BARS="1")
+root = "/comfyui/models"
+targets = [
+    (os.environ["KREA2_MODEL_URL"], f"{root}/unet/{os.environ['KREA2_MODEL_FILENAME']}"),
+    ("https://huggingface.co/Comfy-Org/Krea-2/resolve/8038ce89b91b042141541ad0fa51b985ca262c5f/text_encoders/qwen3vl_4b_fp8_scaled.safetensors",
+     f"{root}/clip/qwen3vl_4b_fp8_scaled.safetensors"),
+    ("https://huggingface.co/Comfy-Org/Krea-2/resolve/8038ce89b91b042141541ad0fa51b985ca262c5f/vae/qwen_image_vae.safetensors",
+     f"{root}/vae/qwen_image_vae.safetensors"),
+]
+for url, dest in targets:
+    with open(LOG, "a") as fh:
+        rc = subprocess.run(["python3", "/opt/prefetch.py", url, dest], stdout=fh, stderr=subprocess.STDOUT, env=env).returncode
+    log(f"prefetch exit {rc} for {os.path.basename(dest)}")
+upload()
+
 log("starting bootstrap")
 upload()
 with open(LOG, "a") as fh: