diff --git a/setup.py b/setup.py
index b401f4d7b8..d5c869b4b4 100644
--- a/setup.py
+++ b/setup.py
@@ -66,7 +66,7 @@
     'slack-sdk<4',
     'mosaicml-cli>=0.6.10,<1',
     'onnx==1.17.0',
-    'onnxruntime==1.19.2',
+    'onnxruntime==1.20.1',
     'boto3>=1.21.45,<2',
     'huggingface-hub>=0.19.0,<0.27',
     'beautifulsoup4>=4.12.2,<5',  # required for model download utils