Commit 0eaa34ec authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix regular empty latent image not working with SD3 and custom sampler.

parent 321e509e
...@@ -380,7 +380,10 @@ class SamplerCustom: ...@@ -380,7 +380,10 @@ class SamplerCustom:
def sample(self, model, add_noise, noise_seed, cfg, positive, negative, sampler, sigmas, latent_image): def sample(self, model, add_noise, noise_seed, cfg, positive, negative, sampler, sigmas, latent_image):
latent = latent_image latent = latent_image
latent_image = latent["samples"] latent_image = latent["samples"]
latent = latent.copy()
latent_image = comfy.sample.fix_empty_latent_channels(model, latent_image) latent_image = comfy.sample.fix_empty_latent_channels(model, latent_image)
latent["samples"] = latent_image
if not add_noise: if not add_noise:
noise = Noise_EmptyNoise().generate_noise(latent) noise = Noise_EmptyNoise().generate_noise(latent)
else: else:
...@@ -539,7 +542,9 @@ class SamplerCustomAdvanced: ...@@ -539,7 +542,9 @@ class SamplerCustomAdvanced:
def sample(self, noise, guider, sampler, sigmas, latent_image): def sample(self, noise, guider, sampler, sigmas, latent_image):
latent = latent_image latent = latent_image
latent_image = latent["samples"] latent_image = latent["samples"]
latent = latent.copy()
latent_image = comfy.sample.fix_empty_latent_channels(guider.model_patcher, latent_image) latent_image = comfy.sample.fix_empty_latent_channels(guider.model_patcher, latent_image)
latent["samples"] = latent_image
noise_mask = None noise_mask = None
if "noise_mask" in latent: if "noise_mask" in latent:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment