From 0eaa34ec5b22fd305159a78ea92b2ef00105ab18 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Wed, 12 Jun 2024 10:32:34 -0400 Subject: [PATCH] Fix regular empty latent image not working with SD3 and custom sampler. --- comfy_extras/nodes_custom_sampler.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/comfy_extras/nodes_custom_sampler.py b/comfy_extras/nodes_custom_sampler.py index 45ef8cf40e1..69f1b94181a 100644 --- a/comfy_extras/nodes_custom_sampler.py +++ b/comfy_extras/nodes_custom_sampler.py @@ -380,7 +380,10 @@ def INPUT_TYPES(s): def sample(self, model, add_noise, noise_seed, cfg, positive, negative, sampler, sigmas, latent_image): latent = latent_image latent_image = latent["samples"] + latent = latent.copy() latent_image = comfy.sample.fix_empty_latent_channels(model, latent_image) + latent["samples"] = latent_image + if not add_noise: noise = Noise_EmptyNoise().generate_noise(latent) else: @@ -539,7 +542,9 @@ def INPUT_TYPES(s): def sample(self, noise, guider, sampler, sigmas, latent_image): latent = latent_image latent_image = latent["samples"] + latent = latent.copy() latent_image = comfy.sample.fix_empty_latent_channels(guider.model_patcher, latent_image) + latent["samples"] = latent_image noise_mask = None if "noise_mask" in latent: