diff --git a/setup.py b/setup.py index 159db91da5..3c34b844a0 100644 --- a/setup.py +++ b/setup.py @@ -57,8 +57,8 @@ 'omegaconf>=2.2.3,<3', 'slack-sdk<4', 'mosaicml-cli>=0.3,<1', - 'onnx==1.13.1', - 'onnxruntime==1.14.1', + 'onnx==1.14.0', + 'onnxruntime==1.15.1', 'cmake>=3.25.0,<=3.26.3', # required for triton-pre-mlir below # PyPI does not support direct dependencies, so we remove this line before uploading from PyPI 'triton-pre-mlir@git+https://github.com/vchiley/triton.git@triton_pre_mlir_sm90#subdirectory=python',