@@ -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 = [
8282inframework = []
8383vllm = [" vllm~=0.17.1" , " pandas" , " timm" ]
8484trtllm = [" 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]
161162prerelease = " allow"
162163
0 commit comments