From 05a6c509fbac9989c067a7b624b3d98646f30dbf Mon Sep 17 00:00:00 2001 From: Nick Date: Tue, 29 Oct 2019 09:29:12 -0600 Subject: [PATCH] update --- generator/ctrl/ctrl_generator.py | 2 +- story/story_data.yaml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index c0b4b88..6bfbe5c 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="Writing Text:", generate_num=45, temperature=0.5, topk=40, nucleus_prob=0): + def __init__(self, control_code="Writing Text: ", generate_num=45, 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 8f9ff4a..3f0d528 100644 --- a/story/story_data.yaml +++ b/story/story_data.yaml @@ -23,4 +23,4 @@ rooms: haunted_hospital: ["lobby", "hallway", "parking", "roof", "pharmacy"] contexts: - apocalypse: "Long ago the bombs fell and the world ended. You are one of the few who is still alive. You are trying to survive by scavenging among the ruins of what is left behind." \ No newline at end of file + apocalypse: "Long ago the bombs fell and the world ended. You are one of the few who is still alive. You are trying to survive by scavenging among the ruins of what is left behind. " \ No newline at end of file