From f94aa3e95eaf87cb8d73953fa9f45f8e6997459f Mon Sep 17 00:00:00 2001
From: xzuyn <16216325+xzuyn@users.noreply.github.com>
Date: Wed, 18 Jun 2025 00:52:07 -0400
Subject: [PATCH] Use `sigma_max`

---
 comfy/samplers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/comfy/samplers.py b/comfy/samplers.py
index 52c9c1fd7..f16782637 100644
--- a/comfy/samplers.py
+++ b/comfy/samplers.py
@@ -464,7 +464,7 @@ def beta_scheduler(model_sampling, steps, alpha=0.6, beta=0.6):
     return torch.FloatTensor(sigs)
 
 def linear_scheduler(model_sampling, steps):
-    return torch.linspace(1.0, 0.0, steps + 1)
+    return torch.linspace(model_sampling.sigma_max, 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):