From c59aadfc10fef6ec2e63896da6208bae6f162b95 Mon Sep 17 00:00:00 2001 From: Nick Date: Thu, 26 Sep 2019 22:52:48 -0600 Subject: [PATCH] update --- console_play.py | 2 +- generator/ctrl/ctrl_generator.py | 2 +- story/story_data.yaml | 2 ++ 3 files changed, 4 insertions(+), 2 deletions(-) diff --git a/console_play.py b/console_play.py index 8d53d6f..0d0cf04 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("peasant") + prompt = get_story_start("fallout") 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 1becd79..33dd72f 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="Books: ", generate_num=80, temperature=0.2): + def __init__(self, control_code="Writing Text: ", generate_num=80, temperature=0.2): self.generate_num=generate_num model_dir = "generator/ctrl/model/seqlen256_v1.ckpt/" diff --git a/story/story_data.yaml b/story/story_data.yaml index d49c052..db35852 100644 --- a/story/story_data.yaml +++ b/story/story_data.yaml @@ -7,6 +7,8 @@ prompts: alien: "You've crash landed on an alien planet. You don't know if you can get your ship working. You get out to explore your surroundings and see " + fallout: "It's been 13 years since the nuclear bombs went off starting this nuclear apocalypse. You're getting low on food and need to go find some. You leave your shelter and see " + action_verbs: classic: ["You attack", "You tell", "You use", "You go"]