diff --git a/pyproject.toml b/pyproject.toml index 7d2b8bad..e650ddd4 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,5 +1,5 @@ [tool.poetry] -name = "fastembed" +name = "fastembed-gpu" version = "0.5.0" description = "Fast, light, accurate library built for retrieval embedding generation" authors = ["Qdrant Team ", "NirantK "] @@ -19,7 +19,7 @@ numpy = [ { version = ">=1.26", python = ">=3.12,<3.13" }, { version = ">=2.1.0", python = ">=3.13" } ] -onnxruntime = [ +onnxruntime-gpu = [ { version = ">=1.17.0,<1.20.0", python = "<3.10" }, { version = ">=1.17.0,!=1.20.0", python = ">=3.10,<3.13" }, { version = ">1.20.0", python = ">=3.13" }