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)