[CLEANUP]

README.md

@@ -192,3 +192,10 @@ for response in responses:
 Swarm Models is released under the [MIT License](https://github.com/The-Swarm-Corporation/swarm-models/LICENSE).
 
 ---
+
+
+# Todo
+
+- [ ] Add cohere models command r
+- [ ] Add gemini and google ai studio
+- [ ] Integrate ollama extensively

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "swarm-models"
-version = "0.1.1"
+version = "0.1.3"
 description = "Swarm Models - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <kye@apac.ai>"]
@@ -23,14 +23,14 @@ classifiers = [
 
 [tool.poetry.dependencies]
 python = "^3.10"
-swarms = "*"
 torch = "*"
 transformers = "*"
 diffusers = "*"
 loguru = "*"
 pydantic = "*"
 langchain-community = "0.0.29"
 together = "*"
+python-dotenv = "*"
 
 
 

swarm_models/__init__.py

@@ -1,6 +1,6 @@
 from swarm_models.base_llm import BaseLLM  # noqa: E402
 from swarm_models.base_multimodal_model import BaseMultiModalModel
-from swarm_models.fuyu import Fuyu  # noqa: E402
+# from swarm_models.fuyu import Fuyu  # noqa: E402
 from swarm_models.gpt4_vision_api import GPT4VisionAPI  # noqa: E402
 from swarm_models.huggingface import HuggingfaceLLM  # noqa: E402
 from swarm_models.idefics import Idefics  # noqa: E402
@@ -46,7 +46,7 @@
 __all__ = [
     "BaseLLM",
     "BaseMultiModalModel",
-    "Fuyu",
+    # "Fuyu",
     "GPT4VisionAPI",
     "HuggingfaceLLM",
     "Idefics",