aligned sample_beam output selection with beam_search (#25375)
* aligned sample_beam specs with beam_search
* pull origin main
* Revert "pull origin main"
This reverts commit 06d356f1137bb52272e120a03636598c44449cf3.
* update test_utils.py
* fix format
* remove comment
---------
Co-authored-by: Shogo Fujita <shogo.fujita@legalontech.jp>