From fa5f2d345adc94ab7814d7fb7061bd48db57dcb8 Mon Sep 17 00:00:00 2001
From: catboxanon <122327233+catboxanon@users.noreply.github.com>
Date: Fri, 2 May 2025 13:49:47 -0400
Subject: [PATCH] Correct beta scheduler to match paper

---
 comfy/samplers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/comfy/samplers.py b/comfy/samplers.py
index 67ae09a25..dbddd2901 100644
--- a/comfy/samplers.py
+++ b/comfy/samplers.py
@@ -445,7 +445,7 @@ def normal_scheduler(model_sampling, steps, sgm=False, floor=False):
 # Implemented based on: https://arxiv.org/abs/2407.12173
 def beta_scheduler(model_sampling, steps, alpha=0.6, beta=0.6):
     total_timesteps = (len(model_sampling.sigmas) - 1)
-    ts = 1 - numpy.linspace(0, 1, steps, endpoint=False)
+    ts = 1 - numpy.linspace(0, 1, steps)
     ts = numpy.rint(scipy.stats.beta.ppf(ts, alpha, beta) * total_timesteps)
 
     sigs = []