Update Storydiffusion_node.py

img2img using more Id image for emb
2024-08-18 17:04:10 +08:00
parent 97ecfd0d3d
commit 96b884d2ec
1 changed files with 1 additions and 1 deletions
@@ -1222,7 +1222,7 @@ def msdiffusion_main(pipe, image_1, image_2, prompts_dual, width, height, steps,
            global write, attn_count
            write = True
            attn_count = 0
-            in_img = image
+            in_img = torch.cat((in_img, image), dim=0)
            torch.cuda.empty_cache()
            if single_files:
                try: