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

This commit is contained in:
comfyanonymous 2023-06-19 15:27:43 -04:00
parent 8883cb0f67
commit 036a22077c

View File

@ -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