Thomas Parnell 4ef95b0f06 [Bugfix] use float32 precision in samplers/test_logprobs.py for comparing with HF (#6409) ...

Signed-off-by: Thomas Parnell <tpa@zurich.ibm.com>

2024-07-15 13:14:49 -04: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

[CI/Test] improve robustness of test (vllm_runner) (#5357)

2024-06-08 08:59:20 +00:00

test_ignore_eos.py

[CI/Test] improve robustness of test (vllm_runner) (#5357)

2024-06-08 08:59:20 +00:00

test_logits_processor.py

[CI/Test] improve robustness of test (vllm_runner) (#5357)

2024-06-08 08:59:20 +00:00

test_logprobs.py

[Bugfix] use float32 precision in samplers/test_logprobs.py for comparing with HF (#6409)

2024-07-15 13:14:49 -04:00

test_ranks.py

[CI/Test] improve robustness of test (vllm_runner) (#5357)

2024-06-08 08:59:20 +00:00

test_rejection_sampler.py

[mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00

test_sampler.py

[CI/Build] Fix Args for _get_logits_warper in Sampler Test (#5922)

2024-06-27 11:43:04 -07:00

test_seeded_generate.py

[CI/Test] improve robustness of test (vllm_runner) (#5357)

2024-06-08 08:59:20 +00:00

test_typical_acceptance_sampler.py

[Speculative Decoding 2/2 ] Integrate typical acceptance sampler into Spec Decode Worker (#5348)

2024-07-01 00:33:05 -07:00