From 71c9680337612bb3b51b5459ead0a2f4d5ae2021 Mon Sep 17 00:00:00 2001 From: Kye Date: Tue, 3 Oct 2023 17:36:00 -0400 Subject: [PATCH] clean up Former-commit-id: 4af54075418ae877cfa0f1ddf563802856d2a5c5 --- mm_agent_example.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mm_agent_example.py b/mm_agent_example.py index 66e050c6..8a7ca8c8 100644 --- a/mm_agent_example.py +++ b/mm_agent_example.py @@ -1,7 +1,7 @@ from swarms.agents import MultiModalAgent load_dict = { - "ImageCaptioning": "cuda:0" + "ImageCaptioning": "cpu" } node = MultiModalAgent(load_dict)