[BUG] [CleanCode] Tuple[int] = (16, 56, 56) in FluxTransformer2DModel

[SlimRG]

Describe the bug

PyLance fixes error.

Выражение типа "tuple[Literal[16], Literal[56], Literal[56]]" не может быть назначено параметру типа "Tuple[int]"
  "tuple[Literal[16], Literal[56], Literal[56]]" невозможно назначить "Tuple[int]"
    Несоответствие размеров tuple: ожидается "1", но получено "3"Pylance

Reproduction

https://github.com/huggingface/diffusers/blob/main/src/diffusers/models/transformers/transformer_flux.py

Go to class FluxTransformer2DModel
Go to def init
See argument: axes_dims_rope: Tuple[int] = (16, 56, 56),

Fix

Need Tuple[int, int, int] = (16, 56, 56),
Or Tuple[int, ...] = (16, 56, 56)

System Info

Python 3.12
Name: diffusers
Version: 0.33.1
Summary: State-of-the-art diffusion in PyTorch and JAX.
Home-page: https://github.com/huggingface/diffusers
Author: The Hugging Face team (past and future) with the help of all our contributors (https://github.com/huggingface/diffusers/graphs/contributors)
Author-email: diffusers@huggingface.co
License: Apache 2.0 License
Location: C:\Users\BBCCA\AppData\Local\Programs\Python\Python312\Lib\site-packages
Requires: filelock, huggingface-hub, importlib-metadata, numpy, Pillow, regex, requests, safetensors
Required-by: compel

Who can help?

No response

[a-r-r-o-w]

Thanks for the suggestion! Would you like to open a PR contribution the improvement?

[sofinvalery]

#11678