From 15db7f42353eef7d66bb16b78225c2f6f588e55d Mon Sep 17 00:00:00 2001 From: ameen-91 Date: Sun, 15 Dec 2024 22:24:51 +0530 Subject: [PATCH] added gpu support --- pyproject.toml | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/pyproject.toml b/pyproject.toml index a598696..691e41a 100644 --- 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"] +