diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index d0cc8e3..142b495 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -273,9 +273,13 @@ class CTRLGenerator(): def generate(self, prompt, options=None): prompt = self.prompt_replace(prompt) - print("\n\n*****DEBUG*****") - print("Prompt is:") - print(prompt + "\n\n") + + debug_print = False + + if debug_print: + print("\n\n*****DEBUG*****") + print("Prompt is:") + print(prompt + "\n\n") if options is None: options = dict() if "used_verbs" not in options: @@ -303,7 +307,8 @@ class CTRLGenerator(): idx = self.generate_next_token(token, tokens_generated, options, num_new_lines, token_num, first_token=first_token, forbid_newline=True) - print(repr(self.idx2word[idx]), end="_") + if debug_print: + print(repr(self.idx2word[idx]), end="_") tokens_generated_so_far = ' '.join([self.idx2word[c] for c in tokens_generated[0][len(text):].squeeze()[:token + 2]]) tokens_generated_so_far = re.sub('(@@ )', '', string=tokens_generated_so_far) @@ -311,6 +316,6 @@ class CTRLGenerator(): result = tokens_generated_so_far token_num += 1 - - print("\n****END DEBUG*****\n") + if debug_print: + print("\n****END DEBUG*****\n") return self.result_replace(result)