From 06b932792f4ace5f28ee7c5956b5cc9eb465e1eb Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Wed, 2 Jul 2025 17:30:35 +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 f3c6635..2368b98 100644
--- a/nodes.py
+++ b/nodes.py
@@ -3964,7 +3964,7 @@ class WanVideoSampler:
 
                         if callback is not None:
                             callback_latent = (latent_model_input.to(device) - noise_pred.to(device) * t.to(device) / 1000).detach().permute(1,0,2,3)
-                            callback(idx, callback_latent, None, iteration_count)                            
+                            callback(iteration_count, callback_latent, None, estimated_iterations)
 
                         # update latent
                         if scheduler == "multitalk":