transformers
43479ef9 - 🚨🚨 Fix group beam search (#24407)

Commit
2 years ago
🚨🚨 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>
Author
Parents
Loading