From 0657dc6960817891efc821da98dffd71f22daaee Mon Sep 17 00:00:00 2001 From: Nick Date: Thu, 26 Sep 2019 16:27:51 -0600 Subject: [PATCH] update --- generator/ctrl/ctrl_generator.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index 3fb7985..574f3e7 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -172,7 +172,7 @@ class CTRLGenerator(): first_letter_capitalized = result[0].isupper() result = result.replace("#", "") result = result.replace("*", "") - #result = first_to_second_person(result) + result = first_to_second_person(result) result = remove_profanity(result) if not first_letter_capitalized: @@ -216,7 +216,8 @@ class CTRLGenerator(): prompt_logits[_token][generated_token] /= self.penalty # disallow some tokens - forbidden_tokens = ['', 'Sco@@', "&@@", "1]@@", "2]@@", "3]@@", "4]@@", "https://www.@@", "[@@", ":@@", "Edit", "&@@", "2:","1:", ":"] + forbidden_tokens = ['', 'Sco@@', "&@@", "1]@@", "2]@@", "3]@@", "4]@@", "https://www.@@", "[@@", ":@@", + "Edit", "&@@", "2:","1:", ":"] if num_new_lines > self.max_new_lines: forbidden_tokens.append("\n")