From a2bb63d546642ef52a03dcb54726efa35b26b29f Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Fri, 28 Feb 2025 13:28:48 +0200
Subject: [PATCH] Update nodes.py

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

diff --git a/nodes.py b/nodes.py
index f3461e1..93e5a8d 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1015,7 +1015,7 @@ class WanVideoSampler:
                 block.self_attn.verbose = True
                 block.self_attn.inner_attention = SparseAttentionMeansim(l1=0.06, pv_l1=0.065)
             if transformer.attention_mode == "spargeattn":
-                saved_state_dict = torch.load("sparge_wan_30_steps_1_iter.pt")
+                saved_state_dict = torch.load("sparge_wan.pt")
                 for key in saved_state_dict.keys():
                     print(key)
                 load_sparse_attention_state_dict(transformer, saved_state_dict, verbose = True)