clean up

Former-commit-id: 4af5407541
1 year ago · 71c9680337
parent 4460f97e56
commit 71c9680337
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)