noise_mask in latent should be in a single format.

This commit is contained in:
comfyanonymous 2023-05-12 20:34:48 -04:00
parent 8ea165dd1e
commit c5c0ea666f
1 changed files with 1 additions and 1 deletions

View File

@ -795,7 +795,7 @@ class SetLatentNoiseMask:
def set_mask(self, samples, mask): def set_mask(self, samples, mask):
s = samples.copy() s = samples.copy()
s["noise_mask"] = mask s["noise_mask"] = mask.reshape((-1, 1, mask.shape[-2], mask.shape[-1]))
return (s,) return (s,)
def common_ksampler(model, seed, steps, cfg, sampler_name, scheduler, positive, negative, latent, denoise=1.0, disable_noise=False, start_step=None, last_step=None, force_full_denoise=False): def common_ksampler(model, seed, steps, cfg, sampler_name, scheduler, positive, negative, latent, denoise=1.0, disable_noise=False, start_step=None, last_step=None, force_full_denoise=False):