transformers
Fix num_heads in _upad_input
#26490
Merged

Loading