Skip to content

GitLab

"docs/vscode:/vscode.git/clone" did not exist on "fad7ca73f8b6b1655b4697628539d937a7e63783"
Commit f44225fd authored by Jedrzej Kosinski's avatar Jedrzej Kosinski
Browse files

Fix infinite while loop being possible in ddim_scheduler

parent 25a4805e
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
comfy/samplers.py
View file @ f44225fd
...@@ -295,7 +295,7 @@ def simple_scheduler(model, steps): ...@@ -295,7 +295,7 @@ def simple_scheduler(model, steps):
def ddim_scheduler(model, steps): def ddim_scheduler(model, steps):
s = model.model_sampling s = model.model_sampling
sigs = [] sigs = []
ss = len(s.sigmas) // steps ss = max(len(s.sigmas) // steps, 1)
x = 1 x = 1
while x < len(s.sigmas): while x < len(s.sigmas):
sigs += [float(s.sigmas[x])] sigs += [float(s.sigmas[x])]
......
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