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( decoded = self.tokenizer.decode(
input_ids[0][self.prompt_length :], input_ids[0][self.prompt_length :],
skip_special_tokens=True, skip_special_tokens=True,
) ) # noqa: E203
if decoded.count(".") > 1: if decoded.count(".") > 1:
return True return True

Loading…
Cancel
Save