From f46abfe4ce3ef33f3c2d1a78dd38abbac7bb18ae Mon Sep 17 00:00:00 2001 From: Valeriy Sofin <112272443+sofinvalery@users.noreply.github.com> Date: Sun, 8 Jun 2025 22:46:30 +0300 Subject: [PATCH] fixed axes_dims_rope init (huggingface#11641) (#11678) --- src/diffusers/models/transformers/transformer_flux.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/models/transformers/transformer_flux.py b/src/diffusers/models/transformers/transformer_flux.py index 7684b60d40..541576b13b 100644 --- a/src/diffusers/models/transformers/transformer_flux.py +++ b/src/diffusers/models/transformers/transformer_flux.py @@ -241,7 +241,7 @@ class FluxTransformer2DModel( joint_attention_dim: int = 4096, pooled_projection_dim: int = 768, guidance_embeds: bool = False, - axes_dims_rope: Tuple[int] = (16, 56, 56), + axes_dims_rope: Tuple[int, int, int] = (16, 56, 56), ): super().__init__() self.out_channels = out_channels or in_channels