From 1e6a7807294f7f420415d319e7e39e7c4f3d374b Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Wed, 14 May 2025 11:40:37 +0300
Subject: [PATCH] Update nodes.py

---
 nodes.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nodes.py b/nodes.py
index 69e835b..51ccceb 100644
--- a/nodes.py
+++ b/nodes.py
@@ -2395,7 +2395,7 @@ class WanVideoSampler:
             sample_scheduler.set_timesteps(steps, device=device, sigmas=sigmas.tolist() if sigmas is not None else None)
         elif 'flowmatch_causvid' in scheduler:
             sample_scheduler = FlowMatchScheduler(num_inference_steps=steps, shift=shift, sigma_min=0, extra_one_step=True)
-            sample_scheduler.timesteps = torch.tensor([1000, 934, 862, 756, 603, 410, 250, 140, 74, 0])[:steps].to(device)
+            sample_scheduler.timesteps = torch.tensor([999, 934, 862, 756, 603, 410, 250, 140, 74])[:steps].to(device)
         
         if timesteps is None:
             timesteps = sample_scheduler.timesteps