PR #37080 [generate] beam search -- fix output cropping

[generate] beam search -- fix output cropping #37080

ArthurZucker merged 6 commits into huggingface:main from gante:beam_search_jagged_beams

handle jagged beams

df6ad6e9

github-actions marked this pull request as draft 343 days ago

better comment

01d9c0c7

bart -- beam search tests print special tokens

86401861

more bart test updates

650f12aa

gante commented on 2025-03-28

more tests!

04ae26ef

gante changed the title ~~[generate] beam search -- handle jagged beams~~ [generate] beam search -- handle jagged output length 343 days ago

gante changed the title ~~[generate] beam search -- handle jagged output length~~ [generate] beam search -- fix output cropping 343 days ago

gante marked this pull request as ready for review 343 days ago

gante requested a review from

zucchini-nlp 343 days ago

better comment

f2f4b808

gante added for patch

ArthurZucker approved these changes on 2025-03-28

ArthurZucker merged 9fd94760 into main 343 days ago

gante deleted the beam_search_jagged_beams branch 343 days ago

Reviewers

ArthurZucker

zucchini-nlp

Assignees

No one assigned

Labels

for patch

Milestone

No milestone