[0.3.1]

examples/models/hf/llama3.py

@@ -1,5 +1,5 @@
 from swarm_models import Agent
-from swarm_models.prompts.finance_agent_sys_prompt import (
+from swarms.prompts.finance_agent_sys_prompt import (
     FINANCIAL_AGENT_SYS_PROMPT,
 )
 import torch

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "swarm-models"
-version = "0.2.7"
+version = "0.3.1"
 description = "Swarm Models - TGSC"
 license = "MIT"
 authors = ["Kye Gomez <[email protected]>"]
@@ -23,7 +23,7 @@ classifiers = [
 
 [tool.poetry.dependencies]
 python = "^3.10"
-torch = "*"
+# torch = "*"
 transformers = "*"
 loguru = "*"
 pydantic = "*"
@@ -32,6 +32,9 @@ python-dotenv = "*"
 litellm = "*"
 tiktoken = "*"
 ollama = "*"
+Pillow = "*"
+termcolor = "*"
+
 
 
 

swarm_models/__init__.py

@@ -1,10 +1,10 @@
 from swarm_models.base_llm import BaseLLM  # noqa: E402
 from swarm_models.base_multimodal_model import BaseMultiModalModel
 from swarm_models.gpt4_vision_api import GPT4VisionAPI  # noqa: E402
-from swarm_models.huggingface import HuggingfaceLLM  # noqa: E402
+# from swarm_models.huggingface import HuggingfaceLLM  # noqa: E402
 # from swarm_models.layoutlm_document_qa import LayoutLMDocumentQA
 from swarm_models.llama3_hosted import llama3Hosted
-from swarm_models.llava import LavaMultiModal  # noqa: E402
+# from swarm_models.llava import LavaMultiModal  # noqa: E402
 # from swarm_models.nougat import Nougat  # noqa: E402
 from swarm_models.openai_embeddings import OpenAIEmbeddings
 from swarm_models.openai_tts import OpenAITTS  # noqa: E402
@@ -32,7 +32,7 @@
     TextModality,
     VideoModality,
 )
-from swarm_models.vilt import Vilt  # noqa: E402
+# from swarm_models.vilt import Vilt  # noqa: E402
 from swarm_models.popular_llms import FireWorksAI
 from swarm_models.openai_function_caller import OpenAIFunctionCaller
 from swarm_models.ollama_model import OllamaModel
@@ -46,9 +46,9 @@
     "BaseLLM",
     "BaseMultiModalModel",
     "GPT4VisionAPI",
-    "HuggingfaceLLM",
+    # "HuggingfaceLLM",
     # "LayoutLMDocumentQA",
-    "LavaMultiModal",
+    # "LavaMultiModal",
     # "Nougat",
     "Palm",
     "OpenAITTS",
@@ -66,7 +66,7 @@
     "MultimodalData",
     "TextModality",
     "VideoModality",
-    "Vilt",
+    # "Vilt",
     "OpenAIEmbeddings",
     "llama3Hosted",
     "FireWorksAI",