From a060b2e6918e1edf1b4a2c67e62098e8ff3a1581 Mon Sep 17 00:00:00 2001 From: nickwalton Date: Thu, 31 Oct 2019 13:41:16 -0600 Subject: [PATCH] fixthethe --- generator/ctrl/ctrl_generator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index 3cd8fec..3f1a32b 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -315,7 +315,7 @@ class CTRLGenerator(): 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 = ' '.join([self.idx2word[c] for c in tokens_generated[0][len(text):token+2].squeeze()[:token + 2]]) 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