From 901aff19fda3ff1ebec6ac4b58c995d9ec94022c Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Wed, 26 Mar 2025 22:39:24 +0200
Subject: [PATCH] Fix euler and dpm....

---
 nodes.py | 15 +++++----------
 1 file changed, 5 insertions(+), 10 deletions(-)

diff --git a/nodes.py b/nodes.py
index f7df7f8..a0e790c 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1726,11 +1726,9 @@ class WanVideoSampler:
                 num_train_timesteps=1000,
                 shift=shift,
                 use_dynamic_shifting=False)
-            sampling_sigmas = get_sampling_sigmas(steps, shift)
-            timesteps, _ = retrieve_timesteps(
-                sample_scheduler,
-                device=device,
-                sigmas=sampling_sigmas)
+            sample_scheduler.set_timesteps(steps, device=device, mu=1)
+            timesteps = sample_scheduler.timesteps
+            
         elif 'dpm++' in scheduler:
             if scheduler == 'dpm++_sde':
                 algorithm_type = "sde-dpmsolver++"
@@ -1741,11 +1739,8 @@ class WanVideoSampler:
                 shift=shift,
                 use_dynamic_shifting=False,
                 algorithm_type= algorithm_type)
-            sampling_sigmas = get_sampling_sigmas(steps, shift)
-            timesteps, _ = retrieve_timesteps(
-                sample_scheduler,
-                device=device,
-                sigmas=sampling_sigmas)
+            sample_scheduler.set_timesteps(steps, device=device, mu=1)
+            timesteps = sample_scheduler.timesteps
         else:
             raise NotImplementedError("Unsupported solver.")