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

Add linear scheduler

Browse Source 
Potentially the scheduler you'd want for wan distill
621d4c322d/lightx2v/models/schedulers/wan/step_distill/scheduler.py (L48)
This commit is contained in:
xzuyn 2025-06-17 22:03:53 -04:00 committed by GitHub
parent d7430c529a
commit 36a76fd68e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/samplers.py
View File

@ -463,6 +463,9 @@ def beta_scheduler(model_sampling, steps, alpha=0.6, beta=0.6):
sigs += [0.0] sigs += [0.0]
return torch.FloatTensor(sigs) return torch.FloatTensor(sigs)
def linear_scheduler(model_sampling, steps):
return torch.linspace(1.0, 0.0, steps + 1)
# from: https://github.com/genmoai/models/blob/main/src/mochi_preview/infer.py#L41 # from: https://github.com/genmoai/models/blob/main/src/mochi_preview/infer.py#L41
def linear_quadratic_schedule(model_sampling, steps, threshold_noise=0.025, linear_steps=None): def linear_quadratic_schedule(model_sampling, steps, threshold_noise=0.025, linear_steps=None):
if steps == 1: if steps == 1:
@ -1046,6 +1049,7 @@ SCHEDULER_HANDLERS = {
"simple": SchedulerHandler(simple_scheduler), "simple": SchedulerHandler(simple_scheduler),
"ddim_uniform": SchedulerHandler(ddim_scheduler), "ddim_uniform": SchedulerHandler(ddim_scheduler),
"beta": SchedulerHandler(beta_scheduler), "beta": SchedulerHandler(beta_scheduler),
"linear": SchedulerHandler(linear_scheduler),
"linear_quadratic": SchedulerHandler(linear_quadratic_schedule), "linear_quadratic": SchedulerHandler(linear_quadratic_schedule),
"kl_optimal": SchedulerHandler(kl_optimal_scheduler, use_ms=False), "kl_optimal": SchedulerHandler(kl_optimal_scheduler, use_ms=False),
} }