mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-06-10 16:15:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
comfyanonymous 2023-06-19 15:27:43 -04:00
parent 8883cb0f67
commit 036a22077c
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/samplers.py
View File

@ -650,7 +650,10 @@ 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
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 k_callback = None
total_steps = len(sigmas) - 1 total_steps = len(sigmas) - 1