From 8b080e943cf5ea0dc7fab2e74f9e4c2c61a499b4 Mon Sep 17 00:00:00 2001 From: AIFSH <149982694+AIFSH@users.noreply.github.com> Date: Fri, 28 Jun 2024 15:01:11 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20studio=5Fnodes.py?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- studio_nodes.py | 23 ++++++++++++----------- 1 file changed, 12 insertions(+), 11 deletions(-) diff --git a/studio_nodes.py b/studio_nodes.py index f7c2ad6..df6d702 100644 --- a/studio_nodes.py +++ b/studio_nodes.py @@ -287,22 +287,23 @@ class DiffutoonNode: cfg_scale,num_inference_steps,animatediff_batch_size,animatediff_stride, vram_limit_level,controlnet1=None,controlnet2=None,controlnet3=None,): global DIFFUTOON_MANAGER + controlnet_path_list = [] + controlnet_model_list = [] + if controlnet1: + controlnet_path_list.append(controlnet1['path']) + controlnet_model_list.append(controlnet1['model']) + if controlnet2: + controlnet_path_list.append(controlnet2['path']) + controlnet_model_list.append(controlnet2['model']) + if controlnet3: + controlnet_path_list.append(controlnet3['path']) + controlnet_model_list.append(controlnet3['model']) + if DIFFUTOON_MANAGER is None: # load models DIFFUTOON_MANAGER = ModelManager(torch_dtype=torch.float16, device=device) shutil.rmtree(os.path.join(textual_inversion_dir,".huggingface"),ignore_errors=True) DIFFUTOON_MANAGER.load_textual_inversions(textual_inversion_dir) - controlnet_path_list = [] - controlnet_model_list = [] - if controlnet1: - controlnet_path_list.append(controlnet1['path']) - controlnet_model_list.append(controlnet1['model']) - if controlnet2: - controlnet_path_list.append(controlnet2['path']) - controlnet_model_list.append(controlnet2['model']) - if controlnet3: - controlnet_path_list.append(controlnet3['path']) - controlnet_model_list.append(controlnet3['model']) DIFFUTOON_MANAGER.load_models([ sd_model_path,