From e1db3026eb8e16bd8df3bf995504eb8e5766da31 Mon Sep 17 00:00:00 2001 From: Kye <kye@apacmediasolutions.com> Date: Tue, 3 Oct 2023 17:36:00 -0400 Subject: [PATCH] clean up Former-commit-id: e457b22968923f0edbce6788d5f7e05533c66f5b --- 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)