Travis Johnson 0258b7a94b

[Bugfix] handle prompt_logprobs in _apply_min_tokens_penalty (#3876) ...

Signed-off-by: Travis Johnson <tsjohnso@us.ibm.com>

2024-04-10 01:39:56 -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

[CI] Try introducing isort. (#3495)

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

test_ranks.py

hotfix isort on logprobs ranks pr (#3622)

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

test_rejection_sampler.py

[CI] Try introducing isort. (#3495)

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

test_sampler.py

[Bugfix] handle prompt_logprobs in _apply_min_tokens_penalty (#3876)

2024-04-10 01:39:56 -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

test_stop_reason.py

[Misc] Include matched stop string/token in responses (#2976)

2024-03-25 17:31:32 -07:00