Pin onnxruntime versions, update deps

Author: istupakov

.devcontainer/cuda/devcontainer-lock.json

@@ -0,0 +1,9 @@
+{
+  "features": {
+    "ghcr.io/devcontainers/features/common-utils:2": {
+      "version": "2.5.8",
+      "resolved": "ghcr.io/devcontainers/features/common-utils@sha256:c42fdefe6d737a3a6f61cc52b23c7c9a565d08cc4d9c303669a7cf2ee5fd81fc",
+      "integrity": "sha256:c42fdefe6d737a3a6f61cc52b23c7c9a565d08cc4d9c303669a7cf2ee5fd81fc"
+    }
+  }
+}

.github/workflows/python-package.yml

@@ -95,20 +95,19 @@ jobs:
           - soundfile: soundfile
           - python-version: "3.10"
             numpy: numpy==1.22.4
-            onnxruntime: onnxruntime==1.18.1
+            onnxruntime: onnxruntime==1.20.1
             hf-hub: huggingface-hub==0.30.2
             typing: typing-extensions==4.6.0
           - python-version: "3.11"
             numpy: numpy~=1.26
-            onnxruntime: onnxruntime~=1.20.1
+            onnxruntime: onnxruntime~=1.22.1
             hf-hub: huggingface-hub~=0.36
           - python-version: "3.12"
             numpy: numpy~=2.0.2
-            onnxruntime: onnxruntime~=1.22.1
+            onnxruntime: onnxruntime~=1.23.2
             hf-hub: huggingface-hub~=1.0.1
           - python-version: "3.13"
             numpy: numpy~=2.3.5
-            onnxruntime: onnxruntime~=1.23.2
           - python-version: "3.13t"
             os: ubuntu-latest
             soundfile: ""
@@ -123,7 +122,7 @@ jobs:
             soundfile: soundfile
           - os: windows-11-arm
             numpy: numpy~=2.3.5
-            onnxruntime: onnxruntime~=1.24.2
+            onnxruntime: onnxruntime~=1.24.4
           - os: windows-11-arm
             python-version: "3.14"
             numpy: ""

README.md

@@ -37,9 +37,6 @@ Key features of **onnx-asr** include:
 > [!NOTE]
 > Supports **Parakeet v2 (En) / v3 (Multilingual)**, **Canary v1/v2 (Multilingual)** and **GigaAM v2/v3 (Ru)** models!
 
-> [!WARNING]
-> onnxruntime 1.24.1 has known compatibility issues with onnx-asr. Please use newer (or older) versions!
-
 > [!TIP]
 > You can check the onnx-asr demo on HF Spaces:
 > 

docs/installation.md

@@ -58,7 +58,7 @@ The supported platforms are primarily determined by available ONNX Runtime wheel
 | Package | Minimum Version |
 |---------|-----------------|
 | numpy | 1.22.4 |
-| onnxruntime | 1.18.1 (or any ONNX Runtime package) |
+| onnxruntime | 1.20.1 (or any ONNX Runtime package) |
 | huggingface-hub | 0.30.2 (optional, for model downloading) |
 | typing-extensions | 4.6.0 (Python < 3.11 only) |
 

pyproject.toml

@@ -39,8 +39,14 @@ classifiers = [
 ]
 
 [project.optional-dependencies]
-cpu = ["onnxruntime>=1.18.1"]
-gpu = ["onnxruntime-gpu>=1.18.1"]
+cpu = [
+    "onnxruntime<1.24; python_version < '3.11'",
+    "onnxruntime>=1.20.1, != 1.24.1, != 1.25.*, != 1.26.0",
+]
+gpu = [
+    "onnxruntime-gpu<1.24; python_version < '3.11'",
+    "onnxruntime-gpu>=1.20.1, != 1.24.1, != 1.25.*, != 1.26.0",
+]
 hub = ["huggingface-hub>=0.30.2"]
 
 [project.urls]