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