cuda

Former-commit-id: aa25e65bad
1 year ago · b8f50de4c4
parent 825b66b44b
commit b8f50de4c4
1 changed files with 1 additions and 1 deletions
--- 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