From a2423997434c110a47090bbe47eb2f39c5902a55 Mon Sep 17 00:00:00 2001 From: wassname Date: Sat, 11 May 2024 14:52:05 +0800 Subject: [PATCH] fix len --- prob_jsonformer/logits_processors.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/prob_jsonformer/logits_processors.py b/prob_jsonformer/logits_processors.py index 6cef012..de88e37 100644 --- a/prob_jsonformer/logits_processors.py +++ b/prob_jsonformer/logits_processors.py @@ -28,11 +28,7 @@ class StringStoppingCriteria(StoppingCriteria): # because of tokens this wont work pefectly gen_ids = input_ids[0][self.prompt_length :] o = self.tokenizer.decode(gen_ids, skip_special_tokens=True) - print("o", o) str_l = len(o) - print( - f"stopping? self.max_length={self.max_length} str_l={str_l},self.prompt_length={self.prompt_length}" - ) if str_l > self.max_length: return True