transformers
fix(generation): stop beam search per-instance when heuristic satisfied
#38778
Merged

fix(generation): stop beam search per-instance when heuristic satisfied #38778

guang-yng
Rocketknight1
zucchini-nlp
guang-yng
gante
gante
gante
gante commented on 2025-06-24
guang-yng
gante
gante
gante commented on 2025-07-07
gante
gante approved these changes on 2025-07-07
gante gante enabled auto-merge (squash) 203 days ago
guang-yng fix(decoding): stop beam search per-instance when heuristic satisfied
a8c62da8
guang-yng Add test case GenerationIntegrationTests.test_beam_search_early_stop_…
8ed13b79
guang-yng Update naming improvement_possibility -> is_early_stop_heuristic_unsa…
43fdb5ab
guang-yng Add comments for early stop heuristic
da8f7db8
gante Update src/transformers/generation/utils.py
04608b6f
gante local make fixup
6de3c599
gante expand test
c942ae7a
gante gante force pushed from a57834df to c942ae7a 203 days ago
gante
HuggingFaceDocBuilderDev
gante fix test?
62ef78c7
gante gante force pushed from e70d6f83 to 62ef78c7 203 days ago
gante Merge branch 'main' into bugfix/heuristic-early-stopping-batch-incons…
e7b19295
gante gante merged 356fd681 into main 203 days ago
gante

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone