transformers

Fix typing hints for different rope parameters per layer type

#43320

Merged