From 611ef4d66b78f6b909ca8622770c89e2985b4265 Mon Sep 17 00:00:00 2001 From: LaVie024 <62406970+LaVie024@users.noreply.github.com> Date: Fri, 2 May 2025 21:37:02 +0000 Subject: [PATCH] Update nodes_custom_sampler.py --- comfy_extras/nodes_custom_sampler.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/comfy_extras/nodes_custom_sampler.py b/comfy_extras/nodes_custom_sampler.py index 3e5be3d3c..e42713347 100644 --- a/comfy_extras/nodes_custom_sampler.py +++ b/comfy_extras/nodes_custom_sampler.py @@ -141,6 +141,7 @@ class BetaSamplingScheduler: "steps": ("INT", {"default": 20, "min": 1, "max": 10000}), "alpha": ("FLOAT", {"default": 0.6, "min": 0.0, "max": 50.0, "step":0.01, "round": False}), "beta": ("FLOAT", {"default": 0.6, "min": 0.0, "max": 50.0, "step":0.01, "round": False}), + "verus": ("BOOLEAN", {"default": False}) } } RETURN_TYPES = ("SIGMAS",) @@ -148,8 +149,8 @@ class BetaSamplingScheduler: FUNCTION = "get_sigmas" - def get_sigmas(self, model, steps, alpha, beta): - sigmas = comfy.samplers.beta_scheduler(model.get_model_object("model_sampling"), steps, alpha=alpha, beta=beta) + def get_sigmas(self, model, steps, alpha, beta, verus): + sigmas = comfy.samplers.beta_scheduler(model.get_model_object("model_sampling"), steps, alpha=alpha, beta=beta, endpoint=verus) return (sigmas, ) class VPScheduler: