Commit 036a2207 authored by comfyanonymous's avatar comfyanonymous
Browse files

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

parent 8883cb0f
...@@ -650,6 +650,9 @@ class KSampler: ...@@ -650,6 +650,9 @@ class KSampler:
self.model_k.latent_image = latent_image self.model_k.latent_image = latent_image
self.model_k.noise = noise self.model_k.noise = noise
if max_denoise:
noise = noise * torch.sqrt(1.0 + sigmas[0] ** 2.0)
else:
noise = noise * sigmas[0] noise = noise * sigmas[0]
k_callback = None k_callback = None
......
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