Fix duplicate sigmas on beta scheduler.

Comfy-Org · Oct 22, 2024 · f82314f · f82314f
1 parent 0075c6d
commit f82314f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/comfy/samplers.py b/comfy/samplers.py
@@ -358,8 +358,11 @@ def beta_scheduler(model_sampling, steps, alpha=0.6, beta=0.6):
     ts = numpy.rint(scipy.stats.beta.ppf(ts, alpha, beta) * total_timesteps)
 
     sigs = []
+    last_t = -1
     for t in ts:
-        sigs += [float(model_sampling.sigmas[int(t)])]
+        if t != last_t:
+            sigs += [float(model_sampling.sigmas[int(t)])]
+        last_t = t
     sigs += [0.0]
     return torch.FloatTensor(sigs)