Fix k_diffusion math being off by a tiny bit during txt2img.

036a2207 · comfyanonymous · 8883cb0f · 036a2207
Commit 036a2207 authored Jun 19, 2023 by comfyanonymous
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

comfy/samplers.py comfy/samplers.py +4 -1

No files found.
--- a/comfy/samplers.py
+++ b/comfy/samplers.py
@@ -650,7 +650,10 @@ class KSampler:
                self.model_k.latent_image = latent_image
                self.model_k.noise = noise

-                noise = noise * sigmas[0]
+                if max_denoise:
+                    noise = noise * torch.sqrt(1.0 + sigmas[0] ** 2.0)
+                else:
+                    noise = noise * sigmas[0]

                k_callback = None
                total_steps = len(sigmas) - 1