xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2026-04-12 00:36:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix Wan enhance a video disabling sageattention patch

Browse Source
This commit is contained in:
kijai 2025-06-06 15:46:33 +03:00
parent f35bb4d336
commit 87b8f85103
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nodes/model_optimization_nodes.py
View File

@ -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,