diff --git a/pyproject.toml b/pyproject.toml index 2a41d70aa2..21d06a9798 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -81,8 +81,8 @@ dependencies = [ [project.optional-dependencies] inframework = [] vllm = ["vllm~=0.10.0", "pandas", "timm"] -trtllm = ["tensorrt-llm>=1.0.0a0,<1.1.0,>=1.0.0rc6", "cuda-python~=12.8.0"] -trt-onnx = ["tensorrt==10.11.0.33", "onnx==1.18.0", "transformers==4.51.3"] +trtllm = ["tensorrt-llm==1.2.0rc0", "cuda-python>=13"] +trt-onnx = ["tensorrt==10.13.0", "onnx==1.18.0", "transformers==4.56.0"] [dependency-groups] # This is a default group so that we install these even with bare `uv sync`