Skip to content

GitLab

"doc/git@developer.sourcefind.cn:wangsen/paddle_dbnet.git" did not exist on "ccb3373d10484c8a5118ae8acc85b5527a0cfc18"
Commit dc198650 authored by comfyanonymous's avatar comfyanonymous
Browse files

sample_dpmpp_2m_sde no longer crashes when step == 1.

parent 4796e615
Hide 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
if sigmas[i + 1] == 0:
# Denoising step
x = denoised
h = None
else:
# DPM-Solver++(2M) SDE
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