🚨🚨 Fix group beam search (#24407)
* group_beam_search now works correctly
* add argument descriptions
* add a comment
* format
* make style
* change comment
* Update src/transformers/generation/beam_search.py
Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>
---------
Co-authored-by: shogo.fujita <shogo.fujita@legalontech.jp>
Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>
