transformers.js
feat: add beam search, diverse beam search, and beam sampling to generate()
#1539
Open
Overview
Commits
6
Changes
View On GitHub

feat: add beam search, diverse beam search, and beam sampling to generate() #1539

justsml wants to merge 6 commits into huggingface:main from justsml:dan/add-beam-support
justsml
justsml justsml changed the title Add beam search support with BeamHypotheses and BeamSearchScorer classes Add beam search decoding to generate() 119 days ago
justsml justsml marked this pull request as ready for review 119 days ago
justsml justsml changed the title Add beam search decoding to generate() Add beam search, diverse beam search, and beam sampling to generate() 119 days ago
justsml justsml changed the title Add beam search, diverse beam search, and beam sampling to generate() feat: add beam search, diverse beam search, and beam sampling to generate() 53 days ago
justsml Add beam search support with BeamHypotheses and BeamSearchScorer classes
b6291bc4
justsml Add diverse beam search support
c162b748
justsml Add beam sampling for beam search
507e6815
justsml Refactor model input expansion for beam search to use index selection
c8271910
justsml Handle past_key_values for beam search in PreTrainedModel to prevent …
f42da6d6
justsml Simplify past_key_values handling in PreTrainedModel for beam search
b0b86bde
justsml justsml force pushed from ad10b094 to b0b86bde 53 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum