Update onnxruntime-gpu version constraints

microsoft · Jan 6, 2025 · 779a8b1 · 779a8b1
1 parent e290c68
commit 779a8b1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/presidio-analyzer/pyproject.toml b/presidio-analyzer/pyproject.toml
@@ -39,7 +39,7 @@ transformers = { version = "*", optional = true }
 huggingface_hub = { version = "*", optional = true }
 gunicorn = {version = "*", optional = true}
 onnxruntime-gpu = { version = "^1.19.1", python = ">=3.10", optional=true}
-onnxruntime = { version = "^1.19.1", python = "<3.10", optional=true}
+onnxruntime-gpu = { version = ">=1.18,<=1.18.1", python = "<3.10", optional=true}
 
 [tool.poetry.extras]
 server = ["flask", "gunicorn"]
@@ -58,6 +58,7 @@ azure-ai-language = [
 gliner = [
     "transformers",
     "huggingface_hub",
+    "onnxruntime-gpu"
     "gliner"
 ]