Skip to content

Commit 9e169f4

Browse files
authored
Import AttentionMixin in transformer_wan_vace.py
Special thanks to @tolgacangoz 🙇‍♂️
1 parent 1f1ce2e commit 9e169f4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/transformers/transformer_wan_vace.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
from ...configuration_utils import ConfigMixin, register_to_config
2222
from ...loaders import FromOriginalModelMixin, PeftAdapterMixin
2323
from ...utils import USE_PEFT_BACKEND, logging, scale_lora_layers, unscale_lora_layers
24-
from ..attention import FeedForward
24+
from ..attention import AttentionMixin, FeedForward
2525
from ..cache_utils import CacheMixin
2626
from ..modeling_outputs import Transformer2DModelOutput
2727
from ..modeling_utils import ModelMixin

0 commit comments

Comments
 (0)