transformers.js
Add beam search, diverse beam search, and beam sampling to generate()
#1539
Open
Overview
Commits
7
Changes
View On GitHub

Add beam search, diverse beam search, and beam sampling to generate() #1539

justsml wants to merge 7 commits into huggingface:main from justsml:dan/add-beam-support
justsml
justsml Add beam search support with BeamHypotheses and BeamSearchScorer classes
fe43e7e5
justsml justsml changed the title Add beam search support with BeamHypotheses and BeamSearchScorer classes Add beam search decoding to generate() 24 days ago
justsml justsml marked this pull request as ready for review 24 days ago
justsml Add diverse beam search support
cc27cb08
justsml Add beam sampling for beam search
2b1f5aec
justsml justsml changed the title Add beam search decoding to generate() Add beam search, diverse beam search, and beam sampling to generate() 24 days ago
justsml Refactor model input expansion for beam search to use index selection
1cad78c0
justsml Handle past_key_values for beam search in PreTrainedModel to prevent …
b9d908e4
justsml Simplify past_key_values handling in PreTrainedModel for beam search
ccf1a43f
justsml Merge branch 'main' into dan/add-beam-support
ad10b094

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