From c7822464c77b7808552ea656e776cb267cde3632 Mon Sep 17 00:00:00 2001 From: somebody Date: Sat, 4 Mar 2023 18:41:48 -0600 Subject: [PATCH] oops --- modeling/stoppers.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/modeling/stoppers.py b/modeling/stoppers.py index b4eeaf51..3ec7a156 100644 --- a/modeling/stoppers.py +++ b/modeling/stoppers.py @@ -132,15 +132,11 @@ class Stoppers: if "newline_in_sequence" not in model.gen_state: model.gen_state["newline_in_sequence"] = [False] * len(input_ids) - print(model.gen_state["newline_in_sequence"]) - for sequence_idx, batch_sequence in enumerate(input_ids): if model.tokenizer.decode(batch_sequence[-1]) == "\n": model.gen_state["newline_in_sequence"][sequence_idx] = True if all(model.gen_state["newline_in_sequence"]): del model.gen_state["newline_in_sequence"] - print("OUT") return True - print("nah its ok") return False