From d4d942c5ee09c221afd566121758633979cac41b Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Mon, 1 Sep 2025 11:17:22 +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 4ce2f0d..d5497ee 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1647,7 +1647,7 @@ class WanVideoScheduler: #WIP
             try:
                 # Plot sigmas and save to a buffer
                 sigmas_np = sample_scheduler.full_sigmas.cpu().numpy()
-                if sample_scheduler.full_sigmas[:-1] == 0:
+                if sample_scheduler.full_sigmas[-1].item() == 0:
                     sigmas_np = sigmas_np[:-1]
                 buf = io.BytesIO()
                 fig = plt.figure(facecolor='#353535')