From 69b87323246cc016f1ddcf24519c5ab3ea0b0fff Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Tue, 3 Oct 2023 17:23:07 -0400
Subject: [PATCH] clean up

---
 mm_agent_example.py                       | 6 +++++-
 swarms/agents/multi_modal_visual_agent.py | 6 +++---
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/mm_agent_example.py b/mm_agent_example.py
index edd1e397..ca1e6051 100644
--- a/mm_agent_example.py
+++ b/mm_agent_example.py
@@ -1,6 +1,10 @@
 from swarms.agents import MultiModalAgent
 
-node = MultiModalAgent()
+load_dict = {
+    "ImageCaptioning": "default_device"
+}
+
+node = MultiModalAgent(load_dict)
 
 text = node.run_text("What is your name? Generate a picture of yourself")
 
diff --git a/swarms/agents/multi_modal_visual_agent.py b/swarms/agents/multi_modal_visual_agent.py
index 454ee8fb..2ea3e5c3 100644
--- a/swarms/agents/multi_modal_visual_agent.py
+++ b/swarms/agents/multi_modal_visual_agent.py
@@ -1637,9 +1637,9 @@ class MultiModalAgent:
         self.langigage = language
 
         # if load_dict is None:
-        load_dict = {
-            "ImageCaptioning": "default_device"
-        }
+        # self.load_dict = {
+        #     "ImageCaptioning": "default_device"
+        # }
 
         self.agent = MultiModalVisualAgent(
             load_dict,