tests/lora/test_layers.py · 96b6f475dda40a0c7d557f73c36fe09c07be2e9c · norm / vllm · GitLab

"megatron/text_generation_server.py" did not exist on "ac3db1595427ace88fd98a5560da6b75c4704a52"

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_layers.py 26.2 KB