From 398f26b90f71fc0b148c1873d72cf34ae3b039ae Mon Sep 17 00:00:00 2001
From: Bogdan Buduroiu <bogdan@buduroiu.com>
Date: Mon, 19 Feb 2024 14:52:14 +0800
Subject: [PATCH] Separate vision dependencies into `vision` extra
---
pyproject.toml | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/pyproject.toml b/pyproject.toml
index d179c185..c067952d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -37,8 +37,9 @@ pillow = { version= "^10.2.0", optional = true}
[tool.poetry.extras]
hybrid = ["pinecone-text"]
fastembed = ["fastembed"]
-local = ["torch", "torchvision", "transformers", "llama-cpp-python", "pillow"]
+local = ["torch", "transformers", "llama-cpp-python"]
pinecone = ["pinecone-client"]
+vision = ["torchvision", "pillow"]
[tool.poetry.group.dev.dependencies]
ipykernel = "^6.25.0"
--
GitLab