From 47bf2b56290d51d996822e029df496d0dfeba72d Mon Sep 17 00:00:00 2001 From: evelynmitchell Date: Wed, 14 Feb 2024 17:24:16 -0700 Subject: [PATCH] flake8 cleanup --- swarms/models/vip_llava.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/swarms/models/vip_llava.py b/swarms/models/vip_llava.py index db532913..4d52877e 100644 --- a/swarms/models/vip_llava.py +++ b/swarms/models/vip_llava.py @@ -83,7 +83,7 @@ class VipLlavaMultiModal(BaseMultiModalModel): ) return self.processor.decode( - generate_ids[0][len(inputs["input_ids"][0]) :], + generate_ids[0][len(inputs["input_ids"][0]):], skip_special_tokens=True, )