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

Correct beta scheduler to match paper

Browse Source
This commit is contained in:
catboxanon 2025-05-02 13:49:47 -04:00
parent d9a87c1e6a
commit fa5f2d345a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/samplers.py
View File

@ -445,7 +445,7 @@ def normal_scheduler(model_sampling, steps, sgm=False, floor=False):
# Implemented based on: https://arxiv.org/abs/2407.12173 # Implemented based on: https://arxiv.org/abs/2407.12173
def beta_scheduler(model_sampling, steps, alpha=0.6, beta=0.6): def beta_scheduler(model_sampling, steps, alpha=0.6, beta=0.6):
total_timesteps = (len(model_sampling.sigmas) - 1) 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) ts = numpy.rint(scipy.stats.beta.ppf(ts, alpha, beta) * total_timesteps)
sigs = [] sigs = []