From b01125e984b3ddcd024ddf5161074a1dcea58d13 Mon Sep 17 00:00:00 2001 From: nickwalton Date: Thu, 31 Oct 2019 13:38:22 -0600 Subject: [PATCH] fixthethe --- generator/ctrl/ctrl_generator.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index d61bfde..3cd8fec 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -310,7 +310,7 @@ class CTRLGenerator(): idx = self.generate_next_token(token, tokens_generated, options, num_new_lines, token_num, first_token=first_token, forbid_newline=True) # assign the token for generation - + tokens_generated[0][token + 1] = idx if debug_print: print(repr(self.idx2word[idx]), end="_") @@ -319,6 +319,7 @@ class CTRLGenerator(): tokens_generated_so_far = re.sub('(@@ )', '', string=tokens_generated_so_far) tokens_generated_so_far = re.sub('(@@ ?$)', '', string=tokens_generated_so_far) result = tokens_generated_so_far + print("\n\nGenerated so far, ", result, "\n\n") token_num += 1 if debug_print: