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: