diffusers dependencies

pyproject.toml

@@ -36,6 +36,7 @@ langchain-experimental = "*"
 playwright = "*"
 duckduckgo-search = "*"
 faiss-cpu = "*"
+diffusers = "*"
 wget = "*"
 httpx = "*"
 ggl = "*"

swarms/models/kosmos_two.py

@@ -20,7 +20,7 @@ class Kosmos:
     """
 
     Args:
-
+    
 
 
     # Initialize Kosmos

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
 
 
 

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."""