Fix infinite while loop being possible in ddim_scheduler

1 year ago · f44225fd5f
parent 25a4805e51
commit f44225fd5f
1 changed files with 1 additions and 1 deletions
--- a/comfy/samplers.py
+++ b/comfy/samplers.py
@ -295,7 +295,7 @@ def simple_scheduler(model, steps):
 def ddim_scheduler(model, steps):
    s = model.model_sampling
    sigs = []
-    ss = len(s.sigmas) // steps
+    ss = max(len(s.sigmas) // steps, 1)
    x = 1
    while x < len(s.sigmas):
        sigs += [float(s.sigmas[x])]