added gpu support

norsulabs · Dec 15, 2024 · 15db7f4 · 15db7f4
1 parent 4d91dec
commit 15db7f4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -38,3 +38,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry.scripts]
 infero = "infero.main:app"
+
+[tool.poetry.extras]
+gpu = ["onnxruntime-gpu"]
+