Nick Hill 81661da7b2

[BugFix] Fix min_tokens when eos_token_id is None (#4389) ...

Co-authored-by: DefTruth <31974251+deftruth@users.noreply.github.com>

2024-04-27 09:52:46 -07:00

..

test_beam_search.py

[Test] Make model tests run again and remove --forked from pytest (#3631)

2024-03-28 21:06:40 -07:00

test_logits_processor.py

[Bugfix] Fix logits processor when prompt_logprobs is not None (#3899)

2024-04-10 00:09:36 -07:00

test_logprobs.py

[Core] Refactoring sampler and support prompt logprob for chunked prefill (#4309)

2024-04-26 13:02:02 +00:00

test_ranks.py

hotfix isort on logprobs ranks pr (#3622)

2024-03-25 11:55:46 -07:00

test_rejection_sampler.py

[Speculative decoding 7/9] Speculative decoding end-to-end correctness tests. (#3951)

2024-04-23 08:02:36 +00:00

test_sampler.py

[BugFix] Fix min_tokens when eos_token_id is None (#4389)

2024-04-27 09:52:46 -07:00

test_seeded_generate.py

[Test] Make model tests run again and remove --forked from pytest (#3631)

2024-03-28 21:06:40 -07:00