Fix duplicate sigmas on beta scheduler.

comfyanonymous · 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)