From 88cd867a48e940b5d05b5f9f8db314a541abeda7 Mon Sep 17 00:00:00 2001 From: Nick Date: Thu, 19 Sep 2019 00:13:12 -0600 Subject: [PATCH] debug --- generator/ctrl/ctrl_generator.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index 0e58604..6a53c2d 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -141,6 +141,9 @@ class CTRLGenerator(): prompt = second_to_first_person(prompt) prompt = self.control_code + prompt + + print("\n\n Prompt to generate by is \n", prompt) + prompt_length = len(prompt) # tokenize provided prompt @@ -252,6 +255,8 @@ class CTRLGenerator(): tokens_generated_so_far = re.sub('(@@ )', '', string=tokens_generated_so_far) tokens_generated_so_far = re.sub('(@@ ?$)', '', string=tokens_generated_so_far) + print(tokens_generated_so_far) + result = tokens_generated_so_far[prompt_length:] except: