From aa25e65bad497eab2268834c886a2c44cf5d613b Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Wed, 18 Oct 2023 16:17:10 -0400
Subject: [PATCH] cuda

---
 swarms/models/fuyu.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/swarms/models/fuyu.py b/swarms/models/fuyu.py
index 4d686b76..d16b2b81 100644
--- a/swarms/models/fuyu.py
+++ b/swarms/models/fuyu.py
@@ -33,7 +33,7 @@ class Fuyu:
     def __init__(
         self,
         pretrained_path: str = "adept/fuyu-8b",
-        device_map: str = "cuda",
+        device_map: str = "cuda:0",
         max_new_tokens: int = 7,
     ):
         self.pretrained_path = pretrained_path