fixed axes_dims_rope init (huggingface#11641) (#11678)

Author: sofinvalery

src/diffusers/models/transformers/transformer_flux.py

@@ -241,7 +241,7 @@ def __init__(
         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