Former-commit-id: e457b22968923f0edbce6788d5f7e05533c66f5b
pull/160/head
Kye 2 years ago
parent 8308ab9ec3
commit e1db3026eb

@ -1,7 +1,7 @@
from swarms.agents import MultiModalAgent from swarms.agents import MultiModalAgent
load_dict = { load_dict = {
"ImageCaptioning": "cuda:0" "ImageCaptioning": "cpu"
} }
node = MultiModalAgent(load_dict) node = MultiModalAgent(load_dict)

Loading…
Cancel
Save