From 05e5b2dbf691d87c02e8deb80f41d9ecd9aede83 Mon Sep 17 00:00:00 2001
From: ExponentialML <59846140+ExponentialML@users.noreply.github.com>
Date: Fri, 15 Mar 2024 11:28:33 -0700
Subject: [PATCH] Fix < 1.0 denoise batch

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

diff --git a/nodes.py b/nodes.py
index ec758f2..a5ae07e 100644
--- a/nodes.py
+++ b/nodes.py
@@ -51,7 +51,7 @@ class ApplyVisualStyle:
         latents = torch.cat([latents] * 2)
 
         if denoise < 1.0:
-            latents[::1] = reference_latent
+            latents[::1] = reference_latent[:1]
         else:
             latents[::2] = reference_latent