diff --git a/pyproject.toml b/pyproject.toml
index d179c185ae90a33076e9eb19dfbad9433452ba09..c067952d9a08f644ff1fbf90d6a8c0ba375ea19a 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"