diff --git a/nodes.py b/nodes.py index 00622ec..9a8cecc 100644 --- a/nodes.py +++ b/nodes.py @@ -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"} \ No newline at end of file +NODE_DISPLAY_NAME_MAPPINGS = {"ApplyVisualStyle": "Apply Visual Style Prompting"}