Cyrus Leung 5ae5ed1e60

[Core] Consolidate prompt arguments to LLM engines (#4328) ...

Co-authored-by: Roger Wang <ywang@roblox.com>

2024-05-28 13:29:31 -07:00

..

__init__.py

[CI/Build] Move test_utils.py to tests/utils.py (#4425)

2024-05-13 23:50:09 +09: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_ignore_eos.py

[Test] Add ignore_eos test (#4519)

2024-05-01 08:45:42 -04:00

test_logits_processor.py

[Core] Consolidate prompt arguments to LLM engines (#4328)

2024-05-28 13:29:31 -07:00

test_logprobs.py

[CI/Build] Move test_utils.py to tests/utils.py (#4425)

2024-05-13 23:50:09 +09:00

test_ranks.py

hotfix isort on logprobs ranks pr (#3622)

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

test_rejection_sampler.py

[Dynamic Spec Decoding] Auto-disable by the running queue size (#4592)

2024-05-08 21:44:00 +00:00

test_sampler.py

[Misc] Remove unnecessary ModelRunner imports (#4703)

2024-05-09 00:17:17 -07:00

test_seeded_generate.py

[Core] Consolidate prompt arguments to LLM engines (#4328)

2024-05-28 13:29:31 -07:00