transformers

Fix `_speculative_sampling` implementation

#28508

Merged