From 87b8f85103c1788bf8c465f02b7ad619a1840d52 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Fri, 6 Jun 2025 15:46:33 +0300
Subject: [PATCH] Fix Wan enhance a video disabling sageattention patch

---
 nodes/model_optimization_nodes.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/nodes/model_optimization_nodes.py b/nodes/model_optimization_nodes.py
index c2e9a9a..682674a 100644
--- a/nodes/model_optimization_nodes.py
+++ b/nodes/model_optimization_nodes.py
@@ -1240,7 +1240,7 @@ Official recommended values https://github.com/ali-vilab/TeaCache/tree/main/TeaC
 
 
 
-from comfy.ldm.modules.attention import optimized_attention
+
 from comfy.ldm.flux.math import apply_rope
 
 def modified_wan_self_attention_forward(self, x, freqs):
@@ -1264,7 +1264,7 @@ def modified_wan_self_attention_forward(self, x, freqs):
 
     feta_scores = get_feta_scores(q, k, self.num_frames, self.enhance_weight)
 
-    x = optimized_attention(
+    x = comfy.ldm.modules.attention.optimized_attention(
         q.view(b, s, n * d),
         k.view(b, s, n * d),
         v,