From d9038d575f780107d2ef907761369bf3fbcbeb2f Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sun, 31 Aug 2025 19:04:01 +0300
Subject: [PATCH] Allow setting both start and end step by sigma

---
 wanvideo/schedulers/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/wanvideo/schedulers/__init__.py b/wanvideo/schedulers/__init__.py
index 236ed53..fcdba1a 100644
--- a/wanvideo/schedulers/__init__.py
+++ b/wanvideo/schedulers/__init__.py
@@ -100,7 +100,7 @@ def get_scheduler(scheduler, steps, start_step, end_step, shift, device, transfo
         timesteps = sample_scheduler.timesteps
 
     steps = len(timesteps)
-    if end_step != -1 and start_step >= end_step:
+    if (isinstance(start_step, int) and end_step != -1 and start_step >= end_step) or (not isinstance(start_step, int) and start_step != -1 and end_step >= start_step):
         raise ValueError("start_step must be less than end_step")
     if denoise_strength < 1.0:
         if start_step != 0: