Update fuyu.py

1 year ago · 0de5ebe089
parent 5a7662f33f
commit 0de5ebe089
1 changed files with 1 additions and 1 deletions
--- a/swarms/models/fuyu.py
+++ b/swarms/models/fuyu.py
@ -39,7 +39,7 @@ class Fuyu(BaseMultiModalModel):
        *args,
        **kwargs,
    ):
-        super().__init__(*args, **kwargs)
+        super().__init__(model_name=model_name, *args, **kwargs)
        self.model_name = model_name
        self.device_map = device_map
        self.max_new_tokens = max_new_tokens