mirror of
https://github.com/wassname/vllm.git
synced 2026-08-15 12:54:49 +08:00
[Build] update requirements of no-device for plugin usage (#12630)
Signed-off-by: Sophie du Couédic <sop@zurich.ibm.com>
This commit is contained in:
@@ -556,7 +556,7 @@ def get_requirements() -> List[str]:
|
||||
return resolved_requirements
|
||||
|
||||
if _no_device():
|
||||
requirements = _read_requirements("requirements-cpu.txt")
|
||||
requirements = _read_requirements("requirements-common.txt")
|
||||
elif _is_cuda():
|
||||
requirements = _read_requirements("requirements-cuda.txt")
|
||||
cuda_major, cuda_minor = torch.version.cuda.split(".")
|
||||
|
||||
Reference in New Issue
Block a user