From 725e68aeea8a4040aea8bf4987770cdcde01c04d Mon Sep 17 00:00:00 2001 From: nickwalton Date: Thu, 31 Oct 2019 13:43:37 -0600 Subject: [PATCH] fixthethe --- generator/ctrl/ctrl_generator.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index 3f1a32b..890a25c 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -315,7 +315,9 @@ 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):token+2].squeeze()[:token + 2]]) + import pdb + pdb.set_trace() + 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) tokens_generated_so_far = re.sub('(@@ ?$)', '', string=tokens_generated_so_far) result = tokens_generated_so_far