diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index 1126a61..8f2ce14 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="Apocalypse ", generate_num=28, temperature=0.6, topk=40, nucleus_prob=0): + def __init__(self, control_code="Apocalypse ", generate_num=28, temperature=0.5, topk=40, nucleus_prob=0): self.generate_num=generate_num model_dir = "generator/ctrl/training_utils/seqlen256_v1.ckpt/" diff --git a/story/story_data.yaml b/story/story_data.yaml index 13a9782..71bce64 100644 --- a/story/story_data.yaml +++ b/story/story_data.yaml @@ -7,7 +7,7 @@ prompts: vague_police: "You are a police officer. You get to work and " - apocalypse: "You are " + apocalypse: "You walk for two hours and take a break. You've left the town you were in and are now in a more rural area. You look around you and see " zombies: "You're on top of a building. You look over the city and see roaming undead everywhere. "