Remove torch and cuda dependencies (#60)

Resolve merge conflicts
landing-ai · Apr 22, 2024 · e803fe1 · e803fe1
1 parent 5f11aea
commit e803fe1
Show file tree

Hide file tree

Showing 11 changed files with 7 additions and 1,088 deletions.
diff --git a/docs/api/data.md b/docs/api/data.md
diff --git a/docs/api/emb.md b/docs/api/emb.md
diff --git a/mkdocs.yml b/mkdocs.yml
@@ -37,7 +37,5 @@ nav:
  - vision_agent.tools: api/tools.md
  - vision_agent.llm: api/llm.md
  - vision_agent.lmm: api/lmm.md
- - vision_agent.data: api/data.md
- - vision_agent.emb: api/emb.md
  - vision_agent.image_utils: api/image_utils.md
  - Old documentation: old.md
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "vision-agent"
-version = "0.1.5"
+version = "0.2.0"
 description = "Toolset for Vision Agent"
 authors = ["Landing AI <[email protected]>"]
 readme = "README.md"
@@ -16,16 +16,12 @@ packages = [{include = "vision_agent"}]
 "documentation" = "https://github.com/landing-ai/vision-agent"
 
 [tool.poetry.dependencies] # main dependency group
-python = ">=3.9,<3.12"
-
+python = ">=3.9"
 numpy = ">=1.21.0,<2.0.0"
 pillow = "10.*"
 requests = "2.*"
 tqdm = ">=4.64.0,<5.0.0"
 pandas = "2.*"
-faiss-cpu = "1.*"
-torch = "2.1.*" # 2.2 causes sentence-transformers to seg fault
-sentence-transformers = "2.*"
 openai = "1.*"
 typing_extensions = "4.*"
 moviepy = "1.*"

diff --git a/tests/test_data.py b/tests/test_data.py
diff --git a/vision_agent/__init__.py b/vision_agent/__init__.py
@@ -1,5 +1,3 @@
 from .agent import Agent
-from .data import DataStore, build_data_store
-from .emb import Embedder, OpenAIEmb, SentenceTransformerEmb, get_embedder
 from .llm import LLM, OpenAILLM
 from .lmm import LMM, LLaVALMM, OpenAILMM, get_lmm
diff --git a/vision_agent/data/__init__.py b/vision_agent/data/__init__.py
diff --git a/vision_agent/data/data.py b/vision_agent/data/data.py
diff --git a/vision_agent/emb/__init__.py b/vision_agent/emb/__init__.py
diff --git a/vision_agent/emb/emb.py b/vision_agent/emb/emb.py