From 7e3ec552f8bf28cd674b7316f5feb1b8008608d1 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Tue, 13 Aug 2024 23:45:00 +0300
Subject: [PATCH] fix download

---
 README.md | 4 ++++
 nodes.py  | 3 ++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index a05f5d3..e8c92ff 100644
--- a/README.md
+++ b/README.md
@@ -2,6 +2,10 @@
 
 These nodes include my wrapper for the original diffusers pipeline, as well as work in progress native ComfyUI implementation.
 
+For the diffusers wrapper models should be downloaded automatically, for the native version you can get the unet here:
+
+https://huggingface.co/Kijai/ControlNeXt-SVD-V2-Comfy/blob/main/controlnext-svd_v2-unet-fp16_converted.safetensors
+
 Diffusers wrapper:
 
 https://github.com/user-attachments/assets/9bc06d4c-b29a-45f5-8a67-211dd5d0f555
diff --git a/nodes.py b/nodes.py
index e5e4400..d92e1e6 100644
--- a/nodes.py
+++ b/nodes.py
@@ -93,7 +93,8 @@ class DownloadAndLoadControlNeXt:
         if not os.path.exists(unet_model_path):
             log.info(f"Downloading model to: {unet_model_path}")
             from huggingface_hub import snapshot_download
-            snapshot_download(repo_id="Pbihao/ControlNeXt", 
+            snapshot_download(repo_id="Kijai/ControlNeXt-SVD-V2-Comfy", 
+                                ignore_patterns=["*converted*"]
                                 local_dir=download_path, 
                                 local_dir_use_symlinks=False)