PR #18851 Generate: get the correct beam index on eos token

Generate: get the correct beam index on eos token #18851

gante merged 2 commits into huggingface:main from gante:batched_beam_search_fix

Get the correct beam index on eos token

297bbf77

gante requested a review from

patrickvonplaten 3 years ago

gante commented on 2022-09-01

Fix test as well

78b2dc14

patrickvonplaten approved these changes on 2022-09-05

gante merged d4dbd7ca into main 3 years ago

gante deleted the batched_beam_search_fix branch 3 years ago

Reviewers

patrickvonplaten

Assignees

No one assigned

Labels

None yet

Milestone

No milestone