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

Add transformer_options to sageattn patch

Browse Source
This commit is contained in:
kijai 2025-09-13 11:04:17 +03:00
parent 36f6fdd7d4
commit 5dc5a15cc4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nodes/model_optimization_nodes.py
View File

@ -114,7 +114,7 @@ class BaseLoaderKJ:
sage_func = set_sage_func(sage_attention) sage_func = set_sage_func(sage_attention)
@torch.compiler.disable() @torch.compiler.disable()
def attention_sage(q, k, v, heads, mask=None, attn_precision=None, skip_reshape=False, skip_output_reshape=False): def attention_sage(q, k, v, heads, mask=None, attn_precision=None, skip_reshape=False, skip_output_reshape=False, transformer_options=None):
if skip_reshape: if skip_reshape:
b, _, _, dim_head = q.shape b, _, _, dim_head = q.shape
tensor_layout="HND" tensor_layout="HND"