Skip to content

Commit 981eb80

Browse files
lgyStoicsayakpaul
andauthored
feat: add qkv projection fuse for longcat transformers (#13021)
feat: add qkv fuse for longcat transformers Co-authored-by: Sayak Paul <[email protected]>
1 parent 1eb40c6 commit 981eb80

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/diffusers/models/transformers/transformer_longcat_image.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
from ...loaders import FromOriginalModelMixin, PeftAdapterMixin
2424
from ...utils import logging
2525
from ...utils.torch_utils import maybe_allow_in_graph
26-
from ..attention import AttentionModuleMixin, FeedForward
26+
from ..attention import AttentionMixin, AttentionModuleMixin, FeedForward
2727
from ..attention_dispatch import dispatch_attention_fn
2828
from ..cache_utils import CacheMixin
2929
from ..embeddings import TimestepEmbedding, Timesteps, apply_rotary_emb, get_1d_rotary_pos_embed
@@ -400,6 +400,7 @@ class LongCatImageTransformer2DModel(
400400
PeftAdapterMixin,
401401
FromOriginalModelMixin,
402402
CacheMixin,
403+
AttentionMixin,
403404
):
404405
"""
405406
The Transformer model introduced in Longcat-Image.

0 commit comments

Comments
 (0)