Travis Johnson c13ad1b7bd

feat: implement the min_tokens sampling parameter (#3124) ...

Signed-off-by: Travis Johnson <tsjohnso@us.ibm.com>
Co-authored-by: Nick Hill <nickhill@us.ibm.com>

2024-03-25 10:14:26 -07:00

..

test_beam_search.py

[Core] Refactor Attention Take 2 (#3462)

2024-03-25 04:39:33 +00:00

test_logprobs.py

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00

test_ranks.py

[Core] Adding token ranks along with logprobs (#3516)

2024-03-25 10:13:10 -07:00

test_rejection_sampler.py

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00

test_sampler.py

feat: implement the min_tokens sampling parameter (#3124)

2024-03-25 10:14:26 -07:00

test_seeded_generate.py

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00