xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-07 04:17:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update nodes_custom_sampler.py

Browse Source
This commit is contained in:
LaVie024 2025-05-02 21:37:02 +00:00 committed by GitHub
parent 80cf00518d
commit 611ef4d66b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy_extras/nodes_custom_sampler.py
View File

@ -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: