3 people 96b6f475dd Remove hardcoded device="cuda" to support more devices (#2503) ...

Co-authored-by: Jiang Li <jiang1.li@intel.com>
Co-authored-by: Kunshang Ji <kunshang.ji@intel.com>

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

..

test_beam_search.py

Align vLLM's beam search implementation with HF generate (#857)

2023-09-04 17:29:42 -07: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