From 1fb193288b7017b0ae6bf5f91adc9b492820e9e0 Mon Sep 17 00:00:00 2001 From: Kye Date: Sat, 4 Nov 2023 21:12:31 -0400 Subject: [PATCH] fuyu fix --- 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 e8d16cdf8..bdd3f904a 100644 --- a/swarms/models/fuyu.py +++ b/swarms/models/fuyu.py @@ -61,6 +61,6 @@ def __call__(self, text: str, img_path: str): model_inputs[k] = v.to(self.device_map) output = self.model.generate( - **model_inputs, max_new_tokens=self.fmax_new_tokens + **model_inputs, max_new_tokens=self.max_new_tokens ) text = self.processor.batch_decode(output[:, -7:], skip_special_tokens=True)