From 351829a2b6f5dea4f2bf41cf6ac770fe54ec1919 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Mon, 29 Dec 2025 15:36:43 +0200
Subject: [PATCH] Fix er_sde steps

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

diff --git a/wanvideo/schedulers/ersde_scheduler.py b/wanvideo/schedulers/ersde_scheduler.py
index 6abd7a7..9574dfb 100644
--- a/wanvideo/schedulers/ersde_scheduler.py
+++ b/wanvideo/schedulers/ersde_scheduler.py
@@ -34,7 +34,7 @@ class ERSDEScheduler():
             sigmas.append(0.0)
         self.sigmas = torch.FloatTensor(sigmas)
         self.sigmas = self.shift * self.sigmas / (1 + (self.shift - 1) * self.sigmas)
-        self.timesteps = self.sigmas * self.num_train_timesteps
+        self.timesteps = self.sigmas[:-1] * self.num_train_timesteps
         self.step_index = 0
         self.old_denoised = None
         self.old_denoised_d = None