tests/samplers/test_sampler.py · dc903e70acf9dba74d6afaa50e7b5650d6b9338a · SIYIXNI / vllm · GitLab

"test/vscode:/vscode.git/clone" did not exist on "51ab3ccf470ac51c8779091dd5db4c91c11e6c8c"

Find file Blame History Permalink

Remove hardcoded `device="cuda" ` to support more devices (#2503) · 96b6f475
Kunshang Ji authored Feb 02, 2024
```
Co-authored-by: Jiang Li <jiang1.li@intel.com>
Co-authored-by: Kunshang Ji <kunshang.ji@intel.com>
```
96b6f475

test_sampler.py 12.6 KB