fix len

2026-09-11 12:44:32 +08:00 · 2024-05-11 14:52:05 +08:00
parent e0feaaf225
commit a242399743
1 changed files with 0 additions and 4 deletions
@@ -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