From 4460f97e566a5762de35bb1e7608df825049ca67 Mon Sep 17 00:00:00 2001 From: Kye Date: Tue, 3 Oct 2023 17:33:12 -0400 Subject: [PATCH] clean up Former-commit-id: 9de553f30b90c7561e1b92433eae4df190a27834 --- mm_agent_example.py | 2 +- swarms/agents/multi_modal_visual_agent.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/mm_agent_example.py b/mm_agent_example.py index ca1e6051..66e050c6 100644 --- a/mm_agent_example.py +++ b/mm_agent_example.py @@ -1,7 +1,7 @@ from swarms.agents import MultiModalAgent load_dict = { - "ImageCaptioning": "default_device" + "ImageCaptioning": "cuda:0" } node = MultiModalAgent(load_dict) diff --git a/swarms/agents/multi_modal_visual_agent.py b/swarms/agents/multi_modal_visual_agent.py index 4f175e22..2ea3e5c3 100644 --- a/swarms/agents/multi_modal_visual_agent.py +++ b/swarms/agents/multi_modal_visual_agent.py @@ -1478,7 +1478,7 @@ class MultiModalVisualAgent: self.models = {} for class_name, device in load_dict.items(): - self.models[class_name] = globals()[class_name]#(device=device) + self.models[class_name] = globals()[class_name](device=device) for class_name, module in globals().items(): if getattr(module, 'template_model', False):