Skip to content

Commit 07b64c8

Browse files
committed
Set onnx < 1.20
Signed-off-by: Charlie Truong <chtruong@nvidia.com>
1 parent dc918fb commit 07b64c8

2 files changed

Lines changed: 37 additions & 122 deletions

File tree

pyproject.toml

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ dependencies = [
6767
"wandb",
6868
"webdataset>=0.2.86",
6969
"nvidia-pytriton ; platform_system != 'Darwin' ",
70-
"flashinfer-python>=0.2.5 ; platform_system != 'Darwin'",
70+
"flashinfer-python>=0.2.5 ; platform_system != 'Darwin' and platform_machine != 'aarch64'",
7171
"Pillow ; platform_system != 'Darwin' and platform_machine != 'aarch64'",
7272
"decord ; platform_system != 'Darwin' and platform_machine != 'aarch64'",
7373
"pyparsing>2.0.2",
@@ -82,7 +82,7 @@ dependencies = [
8282
inframework = []
8383
vllm = ["vllm~=0.17.1", "pandas", "timm"]
8484
trtllm = ["tensorrt-llm~=1.2.0", "cuda-python~=13.1.0"]
85-
trt-onnx = ["tensorrt==10.15.1.29", "onnx==1.20.1", "onnxscript>=0.6.0", "transformers==4.51.3"]
85+
trt-onnx = ["tensorrt==10.15.1.29", "onnx==1.19.1", "onnxscript>=0.6.0", "transformers==4.51.3"]
8686

8787
[dependency-groups]
8888
# This is a default group so that we install these even with bare `uv sync`
@@ -157,6 +157,7 @@ override-dependencies = [
157157
"protobuf~=6.33.5",
158158
"opencv-python-headless; sys_platform == 'never'",
159159
"cryptography>=43.0.0,<47",
160+
"onnx<1.20.0"
160161
]
161162
prerelease = "allow"
162163

0 commit comments

Comments
 (0)