Allow batched images.

2024-03-14 19:33:31 -07:00
parent 79a167a360
commit 04bcf59097
1 changed files with 2 additions and 2 deletions
@@ -46,7 +46,7 @@ class ApplyVisualStyle:
        latents = torch.zeros_like(reference_latent)
        latents = torch.cat([latents] * 2)

-        latents[0] = reference_latent
+        latents[::2] = reference_latent
        denoise_mask = torch.ones_like(latents)[:, :1, ...]
        denoise_mask[0] = 0.

@@ -54,4 +54,4 @@ class ApplyVisualStyle:

 NODE_CLASS_MAPPINGS = {"ApplyVisualStyle": ApplyVisualStyle}

-NODE_DISPLAY_NAME_MAPPINGS = {"ApplyVisualStyle": "Apply Visual Style Prompting"}
+NODE_DISPLAY_NAME_MAPPINGS = {"ApplyVisualStyle": "Apply Visual Style Prompting"}