From b162e76ca0859f7f8b6cb69901ae38aef015aa61 Mon Sep 17 00:00:00 2001 From: evelynmitchell Date: Wed, 27 Nov 2024 19:04:02 -0700 Subject: [PATCH] whitespace colon suppress --- swarms/tools/logits_processor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/swarms/tools/logits_processor.py b/swarms/tools/logits_processor.py index 1c7c8596..40ea3ee8 100644 --- a/swarms/tools/logits_processor.py +++ b/swarms/tools/logits_processor.py @@ -48,7 +48,7 @@ class NumberStoppingCriteria(StoppingCriteria): scores: torch.FloatTensor, ) -> bool: decoded = self.tokenizer.decode( - input_ids[0][self.prompt_length :], + input_ids[0][self.prompt_length :], # noqa: E203 skip_special_tokens=True, ) # noqa: E203