From edbe62514e0da9c56701751cf11ef5c386b25528 Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Fri, 20 Oct 2023 01:22:29 -0400
Subject: [PATCH] diffusers dependencies

---
 pyproject.toml              | 1 +
 swarms/models/kosmos_two.py | 2 +-
 swarms/models/zephyr.py     | 4 ++++
 swarms/tools/tool.py        | 8 --------
 4 files changed, 6 insertions(+), 9 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 441cafc7..dbe378dd 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -36,6 +36,7 @@ langchain-experimental = "*"
 playwright = "*"
 duckduckgo-search = "*"
 faiss-cpu = "*"
+diffusers = "*"
 wget = "*"
 httpx = "*"
 ggl = "*"
diff --git a/swarms/models/kosmos_two.py b/swarms/models/kosmos_two.py
index b36affcb..91118c77 100644
--- a/swarms/models/kosmos_two.py
+++ b/swarms/models/kosmos_two.py
@@ -20,7 +20,7 @@ class Kosmos:
     """
 
     Args:
-
+    
 
 
     # Initialize Kosmos
diff --git a/swarms/models/zephyr.py b/swarms/models/zephyr.py
index 02641a96..8ee12ed9 100644
--- a/swarms/models/zephyr.py
+++ b/swarms/models/zephyr.py
@@ -10,6 +10,10 @@ class Zephyr:
 
 
     Args:
+        max_new_tokens(int) = Number of max new tokens 
+        temperature(float) = temperature of the LLM
+        top_k(float) = top k of the model set to 50
+        top_p(float) = top_p of the model set to 0.95
 
 
 
diff --git a/swarms/tools/tool.py b/swarms/tools/tool.py
index 9bdf45d5..8f01ac0d 100644
--- a/swarms/tools/tool.py
+++ b/swarms/tools/tool.py
@@ -578,14 +578,6 @@ class Tool(BaseTool):
         )
 
 
-class EdgeGPTTool:
-    def __init__(self, model):
-        self.model = model
-
-    def run(self, prompt):
-        return self.model.ask(prompt)
-
-
 class StructuredTool(BaseTool):
     """Tool that can operate on any number of inputs."""