From 09ccffa44e6015533c31f4cb4299060c5f46af47 Mon Sep 17 00:00:00 2001 From: Nick Date: Thu, 19 Sep 2019 22:51:36 -0600 Subject: [PATCH] update --- generator/ctrl/model/generation.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/generator/ctrl/model/generation.py b/generator/ctrl/model/generation.py index 45ce64e..1df4f53 100644 --- a/generator/ctrl/model/generation.py +++ b/generator/ctrl/model/generation.py @@ -270,7 +270,8 @@ while True: tokens_generated_so_far = ' '.join([idx2word[c] for c in tokens_generated[0].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) + tokens_generated_so_far = re.sub('(@@ ?$)', '', string=tokens_generated_so_far) + tokens_generated_so_far = re.sub('(\n\n)', '', string=tokens_generated_so_far) print(tokens_generated_so_far) print()