From cac623ad302d0398ea5d69bc122be85ab5c063f9 Mon Sep 17 00:00:00 2001
From: smthemex <138738845+smthemex@users.noreply.github.com>
Date: Mon, 28 Oct 2024 08:46:24 +0800
Subject: [PATCH] Update Storydiffusion_node.py

---
 Storydiffusion_node.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/Storydiffusion_node.py b/Storydiffusion_node.py
index 61bda1b..9c93bc0 100644
--- a/Storydiffusion_node.py
+++ b/Storydiffusion_node.py
@@ -1847,7 +1847,7 @@ class Storydiffusion_Sampler:
                     gc.collect()
                     torch.cuda.empty_cache()
                     controlnet_path=None
-                    pipe=story_maker_loader(clip_load,clip_vision_path,dir_path,ckpt_path,face_adapter,UniPCMultistepScheduler,controlnet_path,low_vram)
+                    pipe=story_maker_loader(clip_load,clip_vision_path,dir_path,ckpt_path,face_adapter,UniPCMultistepScheduler,controlnet_path,lora_scale,low_vram)
                 mask_image_1=input_id_img_s_dict[role_name_list[0]][0]
                 mask_image_2=input_id_img_s_dict[role_name_list[1]][0]
                 
@@ -1936,8 +1936,11 @@ class Storydiffusion_Sampler:
         else:
             image_list = narry_list(image_pil_list)
         image = torch.from_numpy(np.fromiter(image_list, np.dtype((np.float32, (height, width, 3)))))
-        if use_storydif:
-            pipe.to("cpu")
+        if use_storydif and not prompts_dual:
+            try:
+               pipe.to("cpu")
+            except:
+                pass
         gc.collect()
         torch.cuda.empty_cache()
         return (image, scene_prompts,)