Update download_models.py

download_models.py

@@ -1,24 +1,28 @@
-import os, urllib.request, pathlib
+import urllib.request
 from pathlib import Path
 
-root = pathlib.Path("ComfyUI/models")
+# Crée les dossiers nécessaires à ComfyUI
+root = Path("ComfyUI/models")
 (root / "checkpoints").mkdir(parents=True, exist_ok=True)
+(root / "controlnet").mkdir(parents=True, exist_ok=True)
+(root / "instantid").mkdir(parents=True, exist_ok=True)
 (root / "loras").mkdir(parents=True, exist_ok=True)
-(Path("ComfyUI/models/instantid")).mkdir(parents=True, exist_ok=True)
-(Path("ComfyUI/models/controlnet")).mkdir(parents=True, exist_ok=True)
 
 def wget(url, dest):
     if not dest.exists():
         print("Downloading", url)
         urllib.request.urlretrieve(url, dest)
+    else:
+        print("Already exists:", dest)
 
+# Téléchargement des modèles
 wget(
     "https://huggingface.co/Super-shuhe/InstantID-FaceID-70K/resolve/main/pytorch_model.bin",
-    Path("ComfyUI/models/instantid/Super-shuheInstantID-FaceID-70K.bin")
+    root / "instantid" / "Super-shuheInstantID-FaceID-70K.bin"
 )
 wget(
     "https://huggingface.co/Super-shuhe/InstantID-FaceID-6M/resolve/main/controlnet/diffusion_pytorch_model.safetensors",
-    Path("ComfyUI/models/controlnet/Super-shuheInstantID-FaceID-6M_Controlnet.safetensors")
+    root / "controlnet" / "Super-shuheInstantID-FaceID-6M_Controlnet.safetensors"
 )
 wget(
     "https://huggingface.co/boopyfloopy/gonzalomoXLFluxPony_v40UnityXLDMD/resolve/main/gonzalomoXLFluxPony_v40UnityXLDMD.safetensors",
@@ -27,4 +31,4 @@ wget(
 wget(
     "https://huggingface.co/AiWise/epiCPhoto-XL-LoRA-Derp2/resolve/main/epiCPhotoXL-LoRA-Derp2.safetensors",
     root / "loras" / "epiCPhotoXL-Derp2.safetensors"
-)
+)