From 3399f8feb3c773cff80c8d3a2931f5b7dc83c0c5 Mon Sep 17 00:00:00 2001
From: Joshua Lochner <admin@xenova.com>
Date: Wed, 15 Mar 2023 18:42:33 +0200
Subject: [PATCH] Relax onnxruntime requirement for conversion (#12)

---
 scripts/convert.py       | 1 -
 scripts/requirements.txt | 2 +-
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/scripts/convert.py b/scripts/convert.py
index 9223d49..408e77d 100644
--- a/scripts/convert.py
+++ b/scripts/convert.py
@@ -142,7 +142,6 @@ def quantize(models_name_or_path, model_type):
             model_output=model,
             per_channel=True,
             reduce_range=True,  # should be the same as per_channel
-            activation_type=QuantType.QUInt8,
 
             weight_type=weight_type,
             optimize_model=False,
diff --git a/scripts/requirements.txt b/scripts/requirements.txt
index a6e43bd..9ebdc80 100644
--- a/scripts/requirements.txt
+++ b/scripts/requirements.txt
@@ -1,5 +1,5 @@
 torch
 transformers
 optimum[onnxruntime]>=1.7.1
-onnxruntime==1.10.0
+onnxruntime
 tqdm