Merge branch 'main' into arm_m1_gpu

h2oai · Oct 10, 2023 · 9576d4e · 9576d4e
2 parents deb7353 + 2965b9d
commit 9576d4e
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 6 deletions.
diff --git a/doctr/models/detection/differentiable_binarization/pytorch.py b/doctr/models/detection/differentiable_binarization/pytorch.py
@@ -7,7 +7,6 @@
 from typing import Any, Callable, Dict, List, Optional
 
 import numpy as np
-import onnxruntime as ort
 import torch
 from openvino.runtime import Core
 from torch import nn

diff --git a/doctr/models/recognition/crnn/pytorch.py b/doctr/models/recognition/crnn/pytorch.py
@@ -8,7 +8,6 @@
 from itertools import groupby
 from typing import Any, Callable, Dict, List, Optional, Tuple
 
-import onnxruntime as ort
 import torch
 from openvino.runtime import Core
 from torch import nn

diff --git a/pyproject.toml b/pyproject.toml
@@ -51,9 +51,7 @@ dependencies = [
     "tqdm>=4.30.0",
     "rapidfuzz>=1.6.0",
     "huggingface-hub>=0.4.0",
-    "openvino==2022.3.0",
-    "onnxruntime-gpu>=1.12.1; sys_platform == 'linux'",
-    "onnxruntime>=1.12.1; sys_platform == 'darwin'",
+    "openvino==2022.3.1"
 ]
 
 [project.optional-dependencies]
@@ -105,7 +103,6 @@ dev = [
     "pytest>=5.3.2",
     "coverage[toml]>=4.5.4",
     "hdf5storage>=0.1.18",
-    "onnxruntime>=1.11.0",
     "requests>=2.20.0",
     # Quality
     "flake8>=3.9.0",