wrong path

utils.py

@@ -95,15 +95,9 @@ def postprocess_image(result: torch.Tensor, im_size: list) -> np.ndarray:
 
 
 def downloadModels():
-    MODEL_PATH = hf_hub_download(
-        repo_id="RunDiffusion/Juggernaut-XL-v6",
-        filename="juggernautXL_version6Rundiffusion.safetensors",
+    MODEL_PATH = snapshot_download(
+        repo_id="RunDiffusion/Juggernaut-XL-v6", allow_patterns="*.safetensors"
     )
-    # MODEL_PATH = hf_hub_download(
-    #     repo_id="lllyasviel/fav_models",
-    #     subfolder="fav",
-    #     filename="juggernautXL_v8Rundiffusion.safetensors",
-    # )
     LAYERS_PATH = snapshot_download(
         repo_id="LayerDiffusion/layerdiffusion-v1", allow_patterns="*.safetensors"
     )
@@ -111,12 +105,10 @@ def downloadModels():
         target_path = Path(f"./ComfyUI/models/layer_model/{file.name}")
         if not target_path.exists():
             os.symlink(file, target_path)
-
-    model_target_path = Path(
-        "./ComfyUI/models/checkpoints/juggernautXL_v8Rundiffusion.safetensors"
-    )
-    if not model_target_path.exists():
-        os.symlink(MODEL_PATH, model_target_path)
+    for model in Path(MODEL_PATH).glob("*.safetensors"):
+        model_target_path = Path(f"./ComfyUI/models/checkpoints/{model.name}")
+        if not model_target_path.exists():
+            os.symlink(model, model_target_path)
 
 
 examples = [