whitespace colon suppress

pull/649/head
evelynmitchell 5 months ago
parent 2e88da7bdd
commit ea649b943d

@ -50,7 +50,7 @@ class NumberStoppingCriteria(StoppingCriteria):
decoded = self.tokenizer.decode(
input_ids[0][self.prompt_length :],
skip_special_tokens=True,
)
) # noqa: E203
if decoded.count(".") > 1:
return True

Loading…
Cancel
Save