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:
@@ -11,7 +11,7 @@ scipy
|
||||
pandas
|
||||
setuptools>=64
|
||||
setuptools-scm>=8
|
||||
torch==2.7.1
|
||||
torch==2.8.0
|
||||
torchvision
|
||||
wheel
|
||||
pandas-stubs
|
||||
@@ -28,6 +28,6 @@ numba
|
||||
# Install torch_npu
|
||||
#--pre
|
||||
#--extra-index-url https://mirrors.huaweicloud.com/ascend/repos/pypi
|
||||
torch-npu==2.7.1
|
||||
torch-npu==2.8.0
|
||||
|
||||
transformers<=4.57.1
|
||||
|
||||
Reference in New Issue
Block a user