From 9e10162e691d2d0e19361d4217915c6c18755422 Mon Sep 17 00:00:00 2001
From: ExponentialML <59846140+ExponentialML@users.noreply.github.com>
Date: Fri, 15 Mar 2024 02:01:04 -0700
Subject: [PATCH] Fix CFG

---
 nodes.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/nodes.py b/nodes.py
index 8d69840..f0f2254 100644
--- a/nodes.py
+++ b/nodes.py
@@ -42,7 +42,8 @@ class ApplyVisualStyle:
                 processor = VisualStyleProcessor(m, enabled=enabled)
                 setattr(m, 'forward', processor.visual_style_forward)
 
-        conditioning_prompt[0][0][1:] = reference_image_prompt[0][0][1:]
+        conditioning_prompt = reference_image_prompt + conditioning_prompt 
+        negative_prompt = negative_prompt * 2 
         latents = torch.zeros_like(reference_latent)
         latents = torch.cat([latents] * 2)