clean up

1 year ago · 4af5407541
parent 9de553f30b
commit 4af5407541
1 changed files with 1 additions and 1 deletions
--- 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)