diff --git a/lmdeploy/pytorch/check_env/triton.py b/lmdeploy/pytorch/check_env/triton.py index 8226404e5..047ab9cb6 100644 --- a/lmdeploy/pytorch/check_env/triton.py +++ b/lmdeploy/pytorch/check_env/triton.py @@ -3,7 +3,7 @@ from .base import BaseChecker -MAX_TRITON_VERSION = '3.2.0' +MAX_TRITON_VERSION = '3.3.1' MIN_TRITON_VERSION = '3.0.0' diff --git a/requirements/runtime_cuda.txt b/requirements/runtime_cuda.txt index 557a8ef2a..fe707c923 100644 --- a/requirements/runtime_cuda.txt +++ b/requirements/runtime_cuda.txt @@ -17,8 +17,8 @@ safetensors sentencepiece shortuuid tiktoken -torch<=2.6.0,>=2.0.0 -torchvision<=0.21.0,>=0.15.0 +torch<=2.7.1,>=2.0.0 +torchvision<=0.22.1,>=0.15.0 transformers -triton<=3.2.0,>=3.0.0; sys_platform == "linux" +triton<=3.3.1,>=3.0.0; sys_platform == "linux" uvicorn