Roy f1c0fc3919 Migrate logits computation and gather to model_runner (#3233)

2024-03-20 23:25:01 +00:00

..

test_beam_search.py

[FIX] Fix beam search test (#2930)

2024-02-20 14:37:39 -08:00

test_logprobs.py

Re-enable the 80 char line width limit (#3305)

2024-03-10 19:49:14 -07:00

test_rejection_sampler.py

Remove hardcoded device="cuda" to support more devices (#2503)

2024-02-01 15:46:39 -08:00

test_sampler.py

Migrate logits computation and gather to model_runner (#3233)

2024-03-20 23:25:01 +00:00

test_seeded_generate.py

Support per-request seed (#2514)

2024-02-21 11:47:00 -08:00