From 360666e179eb5ee83e4fec911a40357c0b809fdf Mon Sep 17 00:00:00 2001 From: Kye Date: Sat, 4 Nov 2023 21:12:31 -0400 Subject: [PATCH] fuyu fix Former-commit-id: 1fb193288b7017b0ae6bf5f91adc9b492820e9e0 --- 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 e8d16cdf..bdd3f904 100644 --- a/swarms/models/fuyu.py +++ b/swarms/models/fuyu.py @@ -61,6 +61,6 @@ class Fuyu: 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)