Skip to content

GitLab

Commit dc198650 authored by comfyanonymous's avatar comfyanonymous
Browse files

sample_dpmpp_2m_sde no longer crashes when step == 1.

parent 4796e615
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
comfy/k_diffusion/sampling.py
View file @ dc198650
...@@ -628,6 +628,7 @@ def sample_dpmpp_2m_sde(model, x, sigmas, extra_args=None, callback=None, disabl ...@@ -628,6 +628,7 @@ def sample_dpmpp_2m_sde(model, x, sigmas, extra_args=None, callback=None, disabl
if sigmas[i + 1] == 0: if sigmas[i + 1] == 0:
# Denoising step # Denoising step
x = denoised x = denoised
h = None
else: else:
# DPM-Solver++(2M) SDE # DPM-Solver++(2M) SDE
t, s = -sigmas[i].log(), -sigmas[i + 1].log() t, s = -sigmas[i].log(), -sigmas[i + 1].log()
......
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