From d90853ab35b2e64f51957dd6195304bb5502535a Mon Sep 17 00:00:00 2001
From: AIFSH <149982694+AIFSH@users.noreply.github.com>
Date: Mon, 5 Aug 2024 14:09:58 +0800
Subject: [PATCH] Update studio_nodes.py

fix controlne loader
---
 studio_nodes.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/studio_nodes.py b/studio_nodes.py
index df6d702..674513a 100644
--- a/studio_nodes.py
+++ b/studio_nodes.py
@@ -116,9 +116,9 @@ class ControlNetPathLoader:
         elif model_id == "lineart_anime":
             filename = "control_v11p_sd15s2_lineart_anime.pth"
         
-        if control_net_name:
+        if control_net_name==filename:
             controlnet_path = folder_paths.get_full_path("controlnet", control_net_name)
-            assert filename == control_net_name, f"{filename} dismatch with {control_net_name},please choose correct model"
+            # assert filename == control_net_name, f"{filename} dismatch with {control_net_name},please choose correct model"
         else:
             controlnet_path = hf_hub_download(repo_id="lllyasviel/ControlNet-v1-1",
         
@@ -302,6 +302,7 @@ class DiffutoonNode:
         if DIFFUTOON_MANAGER is None:
             # load models
             DIFFUTOON_MANAGER = ModelManager(torch_dtype=torch.float16, device=device)
+            shutil.rmtree(os.path.join(textual_inversion_dir,".cache"),ignore_errors=True)
             shutil.rmtree(os.path.join(textual_inversion_dir,".huggingface"),ignore_errors=True)
             DIFFUTOON_MANAGER.load_textual_inversions(textual_inversion_dir)
             
@@ -353,4 +354,4 @@ class DiffutoonNode:
         outfile = os.path.join(output_dir,f'{time.time_ns()}_shaded' + os.path.basename(source_video_path))
         save_video(output_video, outfile, fps=fps)
         DIFFUTOON_MANAGER.to("cpu")
-        return (outfile,)
\ No newline at end of file
+        return (outfile,)