Update download_models.py

download_models.py

@@ -1,27 +1,40 @@
-import os, urllib.request, pathlib
+import os
+import urllib.request
+from pathlib import Path
 
-root = pathlib.Path("models")
+root = Path("ComfyUI/models")
+(root / "instantid").mkdir(parents=True, exist_ok=True)
+(root / "controlnet").mkdir(parents=True, exist_ok=True)
 (root / "checkpoints").mkdir(parents=True, exist_ok=True)
 (root / "loras").mkdir(parents=True, exist_ok=True)
 
 def wget(url, dest):
     if not dest.exists():
-        print("Downloading", url)
+        print(f"Downloading {url}")
         urllib.request.urlretrieve(url, dest)
-        
+    else:
+        print(f"Already exists: {dest}")
+
+# --- InstantID ---
 wget(
-    "https://huggingface.co/Super-shuhe/InstantID-FaceID-70K/blob/main/pytorch_model.bin",
-    Path("ComfyUI/models/instantid/Super-shuheInstantID-FaceID-6M.bin"
+    "https://huggingface.co/Super-shuhe/InstantID-FaceID-70K/resolve/main/pytorch_model.bin",
+    root / "instantid" / "Super-shuheInstantID-FaceID-6M.bin"
 )
+
+# --- ControlNet ---
 wget(
-    "https://huggingface.co/Super-shuhe/InstantID-FaceID-6M/blob/main/controlnet/diffusion_pytorch_model.safetensors",
-    Path("ComfyUI/models/controlnet/Super-shuheInstantID-FaceID-6M Controlnet.safetensors"
+    "https://huggingface.co/Super-shuhe/InstantID-FaceID-6M/resolve/main/controlnet/diffusion_pytorch_model.safetensors",
+    root / "controlnet" / "Super-shuheInstantID-FaceID-6M Controlnet.safetensors"
 )
+
+# --- Checkpoint ---
 wget(
-    "https://huggingface.co/boopyfloopy/gonzalomoXLFluxPony_v40UnityXLDMD/blob/main/gonzalomoXLFluxPony_v40UnityXLDMD.safetensors",
+    "https://huggingface.co/boopyfloopy/gonzalomoXLFluxPony_v40UnityXLDMD/resolve/main/gonzalomoXLFluxPony_v40UnityXLDMD.safetensors",
     root / "checkpoints" / "gonzalomoXLFluxPony_v40UnityXLDMD.safetensors"
 )
+
+# --- LoRA ---
 wget(
-    "https://huggingface.co/AiWise/epiCPhoto-XL-LoRA-Derp2/blob/main/epiCPhotoXL-LoRA-Derp2.safetensors",
+    "https://huggingface.co/AiWise/epiCPhoto-XL-LoRA-Derp2/resolve/main/epiCPhotoXL-LoRA-Derp2.safetensors",
     root / "loras" / "epiCPhotoXL-Derp2.safetensors"
-)
+)