nodes/ExponentialML-ComfyUI_VisualStylePrompting
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Allow batched images.

Browse Source
This commit is contained in:
ExponentialML
2024-03-14 19:33:31 -07:00
committed by GitHub
parent 79a167a360
commit 04bcf59097
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+1 -1
View File
@@ -46,7 +46,7 @@ class ApplyVisualStyle:
latents = torch.zeros_like(reference_latent) latents = torch.zeros_like(reference_latent)
latents = torch.cat([latents] * 2) latents = torch.cat([latents] * 2)
latents[0] = reference_latent latents[::2] = reference_latent
denoise_mask = torch.ones_like(latents)[:, :1, ...] denoise_mask = torch.ones_like(latents)[:, :1, ...]
denoise_mask[0] = 0. denoise_mask[0] = 0.