Upload download_models.py with huggingface_hub

download_models.py

@@ -360,15 +360,25 @@ try:
     # Register additional model folders in ComfyUI
     folder_paths_file = "/app/ComfyUI/extra_model_paths.yaml"
     with open(folder_paths_file, "w") as f:
-        f.write(f"""
-evaclip: {MODELS_DIR}/evaclip
-insightface: {MODELS_DIR}/insightface
-pulid: {MODELS_DIR}/pulid
-checkpoints: {MODELS_DIR}/checkpoints
-controlnet: {MODELS_DIR}/controlnet
-ipadapter: {MODELS_DIR}/ipadapter
-clip_vision: {MODELS_DIR}/clip_vision
-""")
+        f.write("""# Extra model paths for ComfyUI
+checkpoints:
+  - {}/checkpoints
+controlnet:
+  - {}/controlnet
+clip_vision:
+  - {}/clip_vision
+ipadapter:
+  - {}/ipadapter
+pulid:
+  - {}/pulid
+evaclip:
+  - {}/evaclip
+insightface:
+  - {}/insightface
+""".format(
+    MODELS_DIR, MODELS_DIR, MODELS_DIR, 
+    MODELS_DIR, MODELS_DIR, MODELS_DIR, MODELS_DIR
+))
     print(f"Created extra_model_paths.yaml at {folder_paths_file}")
     
     # Link from ComfyUI's default directories to our writable directory