Update Storydiffusion_node.py

2024-10-28 08:46:24 +08:00
parent 05564b7889
commit cac623ad30
1 changed files with 6 additions and 3 deletions
@@ -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,)