Zhuohan Li 63e2a6419d [FIX] Fix beam search test (#2930)

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

..

test_beam_search.py

[FIX] Fix beam search test (#2930)

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

test_logprobs.py

[CI] Add Buildkite (#2355)

2024-01-14 12:37:58 -08: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

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

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