Skip to content

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

Open
justsml wants to merge 6 commits into
huggingface:mainfrom
justsml:dan/add-beam-support
Open

feat: add beam search, diverse beam search, and beam sampling to generate()#1539
justsml wants to merge 6 commits into
huggingface:mainfrom
justsml:dan/add-beam-support

Simplify past_key_values handling in PreTrainedModel for beam search

b0b86bd
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs