Merge pull request #4 from ExponentialML/update/fix-cfg

Fix CFG
2024-03-15 02:01:54 -07:00
parent 36f6b86835 9e10162e69
commit 0bda86efab
1 changed files with 2 additions and 1 deletions
@@ -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)