From 36a76fd68e291dbf0f0f22e70553219bb7eb9803 Mon Sep 17 00:00:00 2001
From: xzuyn <16216325+xzuyn@users.noreply.github.com>
Date: Tue, 17 Jun 2025 22:03:53 -0400
Subject: [PATCH] Add `linear` scheduler

Potentially the scheduler you'd want for wan distill
https://github.com/ModelTC/lightx2v/blob/621d4c322dded70e9f9cd8f58fcce25d911253ce/lightx2v/models/schedulers/wan/step_distill/scheduler.py#L48
---
 comfy/samplers.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/comfy/samplers.py b/comfy/samplers.py
index efe9bf867..52c9c1fd7 100644
--- a/comfy/samplers.py
+++ b/comfy/samplers.py
@@ -463,6 +463,9 @@ def beta_scheduler(model_sampling, steps, alpha=0.6, beta=0.6):
     sigs += [0.0]
     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
 def linear_quadratic_schedule(model_sampling, steps, threshold_noise=0.025, linear_steps=None):
     if steps == 1:
@@ -1046,6 +1049,7 @@ SCHEDULER_HANDLERS = {
     "simple": SchedulerHandler(simple_scheduler),
     "ddim_uniform": SchedulerHandler(ddim_scheduler),
     "beta": SchedulerHandler(beta_scheduler),
+    "linear": SchedulerHandler(linear_scheduler),
     "linear_quadratic": SchedulerHandler(linear_quadratic_schedule),
     "kl_optimal": SchedulerHandler(kl_optimal_scheduler, use_ms=False),
 }