Skip to content

Commit 27724f5

Browse files
committed
new: gpu package (#224)
1 parent 6ca7561 commit 27724f5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pyproject.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
[tool.poetry]
2-
name = "fastembed"
2+
name = "fastembed-gpu"
33
version = "0.4.2"
44
description = "Fast, light, accurate library built for retrieval embedding generation"
55
authors = ["Qdrant Team <[email protected]>", "NirantK <[email protected]>"]
@@ -13,7 +13,7 @@ keywords = ["vector", "embedding", "neural", "search", "qdrant", "sentence-trans
1313
[tool.poetry.dependencies]
1414
python = ">=3.8.0,<3.13"
1515
onnx = "^1.15.0"
16-
onnxruntime = ">=1.17.0,<1.20.0"
16+
onnxruntime-gpu = ">=1.17.0,<1.20.0"
1717
tqdm = "^4.66"
1818
requests = "^2.31"
1919
tokenizers = ">=0.15,<1.0"

0 commit comments

Comments
 (0)