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

NeoWorldTeam · Jun 12, 2024 · 0eaa34e · 0eaa34e
1 parent 321e509
commit 0eaa34e
Showing 1 changed file with 5 additions and 0 deletions.
diff --git 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: