clean up

Former-commit-id: e457b22968923f0edbce6788d5f7e05533c66f5b
2 years ago · e1db3026eb
parent 8308ab9ec3
commit e1db3026eb
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)