diff --git a/console_play.py b/console_play.py index 03741a3..4deccf5 100644 --- a/console_play.py +++ b/console_play.py @@ -23,7 +23,7 @@ def console_print(str, pycharm=False): def play_unconstrained(): generator = CTRLGenerator() #generator = WebGenerator(CRED_FILE) - prompt = get_story_start("forest") + prompt = get_story_start("hospital") story_manager = UnconstrainedStoryManager(generator) story_manager.start_new_story(prompt) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index 74c5563..d839797 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -20,7 +20,7 @@ def loss(labels, logits): class CTRLGenerator(): - def __init__(self, control_code="Horror Text: ", generate_num=80, temperature=0.4): + def __init__(self, control_code="Books: ", generate_num=80, temperature=0.4): self.generate_num=generate_num model_dir = "generator/ctrl/model/seqlen256_v1.ckpt/" @@ -158,7 +158,7 @@ class CTRLGenerator(): if prompt[-1] != " ": prompt = prompt + " " - #prompt = second_to_first_person(prompt) + prompt = second_to_first_person(prompt) prompt = self.control_code + prompt # print("\n\nAFTER PROMPT_REPLACE") @@ -218,7 +218,7 @@ class CTRLGenerator(): # disallow some tokens forbidden_tokens = ['', 'Sco@@', "&@@", "1]@@", "2]@@", "3]@@", "4]@@", "https://www.@@", "[@@", ":@@", "Edit", "&@@", "2:","1:", ":", "Edit@@", "EDI@@", "EDIT@@", "edit", "TL@@", "tl@@", ";@@", - '**', "http://@@", "Redd@@", "UP@@"] + '**', "http://@@", "Redd@@", "UP@@", "mom"] if num_new_lines > self.max_new_lines: forbidden_tokens.append("\n")