upgrade torch npu version (#4433)
vLLM graph feature now rely on torch >=2.8. To make graph mode work, we need upgrade torch version as well. For long term support, upgrade torch to a newer one is good to go as well. Related vLLM change: https://github.com/vllm-project/vllm/pull/25110 - vLLM version: v0.11.2 - vLLM main: https://github.com/vllm-project/vllm/commit/v0.11.2
This commit is contained in:
@@ -18,8 +18,8 @@ requires = [
|
||||
"setuptools>=64",
|
||||
"setuptools-scm>=8",
|
||||
"transformers<=4.57.1",
|
||||
"torch-npu==2.7.1",
|
||||
"torch==2.7.1",
|
||||
"torch-npu==2.8.0",
|
||||
"torch==2.8.0",
|
||||
"torchvision",
|
||||
"wheel",
|
||||
"msgpack",
|
||||
|
||||
Reference in New Issue
Block a user