transformers

Fix `max_length_q` and `max_length_k` types to `flash_attn_varlen_func`

#37206

Merged