Mayank Agarwal c4e32e206f Add support for beam search's num_return_sequencs flag in flax (#23082) ...

* add code for numReturnSeq

* add flax support for num return sequences

* Make Fix up for changes

* add test for num return sequences

* lint

2023-05-03 10:50:34 -04:00

12 KiB

Raw Blame History

View Raw