From 96b884d2ecfbec218a52db741c70bceef257d4c6 Mon Sep 17 00:00:00 2001
From: smthemex <138738845+smthemex@users.noreply.github.com>
Date: Sun, 18 Aug 2024 17:04:10 +0800
Subject: [PATCH] Update Storydiffusion_node.py

img2img using more Id image for emb
---
 Storydiffusion_node.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Storydiffusion_node.py b/Storydiffusion_node.py
index 1058c7d..e023da2 100644
--- a/Storydiffusion_node.py
+++ b/Storydiffusion_node.py
@@ -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: