From 0de5ebe08908e666675cb1a067ff242da7692cf4 Mon Sep 17 00:00:00 2001 From: Eternal Reclaimer <98760976+kyegomez@users.noreply.github.com> Date: Sun, 10 Dec 2023 11:50:06 -0800 Subject: [PATCH] Update fuyu.py --- 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 862ed882..f722bbb6 100644 --- 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