Fix < 1.0 denoise batch

2024-03-15 11:28:33 -07:00
parent e18592574d
commit 05e5b2dbf6
1 changed files with 1 additions and 1 deletions
@@ -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