Skip to content

Commit fb531ec

Browse files
authored
Add AttentionMixin to WanVACETransformer3DModel
to enable methods like `set_attn_processor()`.
1 parent 9e4a75b commit fb531ec

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/diffusers/models/transformers/transformer_wan_vace.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,9 @@ def forward(
134134
return conditioning_states, control_hidden_states
135135

136136

137-
class WanVACETransformer3DModel(ModelMixin, ConfigMixin, PeftAdapterMixin, FromOriginalModelMixin, CacheMixin):
137+
class WanVACETransformer3DModel(
138+
ModelMixin, ConfigMixin, PeftAdapterMixin, FromOriginalModelMixin, CacheMixin, AttentionMixin
139+
):
138140
r"""
139141
A Transformer model for video-like data used in the Wan model.
140142

0 commit comments

Comments
 (0)