Skip to content
Snippets Groups Projects
Unverified Commit 398f26b9 authored by Bogdan Buduroiu's avatar Bogdan Buduroiu
Browse files

Separate vision dependencies into `vision` extra

parent 7e39ee49
No related branches found
No related tags found
No related merge requests found
...@@ -37,8 +37,9 @@ pillow = { version= "^10.2.0", optional = true} ...@@ -37,8 +37,9 @@ pillow = { version= "^10.2.0", optional = true}
[tool.poetry.extras] [tool.poetry.extras]
hybrid = ["pinecone-text"] hybrid = ["pinecone-text"]
fastembed = ["fastembed"] fastembed = ["fastembed"]
local = ["torch", "torchvision", "transformers", "llama-cpp-python", "pillow"] local = ["torch", "transformers", "llama-cpp-python"]
pinecone = ["pinecone-client"] pinecone = ["pinecone-client"]
vision = ["torchvision", "pillow"]
[tool.poetry.group.dev.dependencies] [tool.poetry.group.dev.dependencies]
ipykernel = "^6.25.0" ipykernel = "^6.25.0"
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment