fuyu fix

kyegomez · Nov 5, 2023 · 1fb1932 · 1fb1932
1 parent 6e6fe8d
commit 1fb1932
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)