From cc4f3a85c62404a68dbc8e636e39fda65b26e952 Mon Sep 17 00:00:00 2001
From: rmittal-github <61574997+rmittal-github@users.noreply.github.com>
Date: Mon, 6 May 2024 14:16:55 +0530
Subject: [PATCH] Fix onnx dependencies in requirements.txt (#43)

---
 requirements.txt | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 044591a..73ed3b6 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,5 +4,7 @@
 nemo_toolkit>=1.6.0
 nvidia-eff>=0.6.4
 nvidia-eff-tao-encryption>=0.1.8
-onnxruntime-gpu>=1.13.1
+onnx==1.14.1
+onnxruntime==1.16.3
+onnxruntime-gpu==1.16.3
 onnx_graphsurgeon