clean up playground

Former-commit-id: a178953a6c
1 year ago · ad544cf908
parent bb971bcb25
commit ad544cf908
21 changed files with 6 additions and 3 deletions
--- a/playground/agents/mm_agent_example.py
+++ b/playground/agents/mm_agent_example.py
--- a/playground/agents/omni_exa_example.py
+++ b/playground/agents/omni_exa_example.py
--- a/playground/swarms/autoscaler.py
+++ b/playground/swarms/autoscaler.py
--- a/playground/swarms/chat.py
+++ b/playground/swarms/chat.py
--- a/playground/swarms/debate.py
+++ b/playground/swarms/debate.py
--- a/playground/swarms/dialogue_simulator.py
+++ b/playground/swarms/dialogue_simulator.py
--- a/playground/swarms/easy_example.py
+++ b/playground/swarms/easy_example.py
--- a/playground/swarms/godmode.py
+++ b/playground/swarms/godmode.py
--- a/playground/swarms/gui_app.py
+++ b/playground/swarms/gui_app.py
--- a/playground/swarms/multi_agent_collab.py
+++ b/playground/swarms/multi_agent_collab.py
--- a/playground/swarms/multi_agent_debate.py
+++ b/playground/swarms/multi_agent_debate.py
--- a/playground/swarms/orchestrate.py
+++ b/playground/swarms/orchestrate.py
--- a/playground/swarms/orchestrator.py
+++ b/playground/swarms/orchestrator.py
--- a/playground/swarms/social_app.py
+++ b/playground/swarms/social_app.py
--- a/playground/swarms/swarms_example.py
+++ b/playground/swarms/swarms_example.py
--- a/playground/swarms/todo_app.py
+++ b/playground/swarms/todo_app.py
--- a/playground/worker/ultranode_example.py
+++ b/playground/worker/ultranode_example.py
--- a/playground/worker/worker_auto.py
+++ b/playground/worker/worker_auto.py
--- a/playground/worker/worker_ultra.py
+++ b/playground/worker/worker_ultra.py
--- a/pyproject.toml
+++ b/pyproject.toml
@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"

 [tool.poetry]
 name = "swarms"
-version = "1.7.6"
+version = "1.7.7"
 description = "Swarms - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <kye@apac.ai>"]
--- a/swarms/models/mistral.py
+++ b/swarms/models/mistral.py
@ -47,9 +47,12 @@ class Mistral:
        task: str
    ):
        """Run the model on a given task."""
-        
+
        try:
-            model_inputs = self.tokenizer([task], return_tensors="pt").to(self.device)
+            model_inputs = self.tokenizer(
+                [task], 
+                return_tensors="pt"
+            ).to(self.device)
            generated_ids = self.model.generate(
                **model_inputs, 
                max_length=self.max_length,