diff --git a/generator/gpt2/gpt2_generator.py b/generator/gpt2/gpt2_generator.py index 4f147c1..fdb0fe7 100644 --- a/generator/gpt2/gpt2_generator.py +++ b/generator/gpt2/gpt2_generator.py @@ -16,7 +16,7 @@ class GPT2Generator: self.top_k = top_k self.top_p = top_p - self.model_name = "model_v4" + self.model_name = "model_v5" self.model_dir = "generator/gpt2/models" self.checkpoint_path = os.path.join(self.model_dir, self.model_name) diff --git a/install.sh b/install.sh index 00573f7..a3514ce 100755 --- a/install.sh +++ b/install.sh @@ -1,6 +1,6 @@ MODELS_DIRECTORY=generator/gpt2/models -MODEL_VERSION=model_v4 -MODEL_NAME=model-200 +MODEL_VERSION=model_v5 +MODEL_NAME=model-550 DOWNLOAD_URL=https://students.cs.byu.edu/~nickwalt if [ -d "${MODELS_DIRECTORY}/${MODEL_VERSION}" ]; then diff --git a/play_dm.py b/play_dm.py index 3947d88..abcacd1 100644 --- a/play_dm.py +++ b/play_dm.py @@ -16,7 +16,10 @@ class AIPlayer: return self.generator.generate_raw(prompt) def play_dm(): - generator = GPT2Generator() + + console_print("Initializing AI Dungeon DM Mode") + generator = GPT2Generator(temperature=0.9) + story_manager = UnconstrainedStoryManager(HumanDM()) context, prompt = select_game() console_print(context + prompt) @@ -31,6 +34,9 @@ def play_dm(): print(action) print("******END DEBUG******\n") action = action.split("\n")[0] + punc = action.rfind(".") + if punc > 0: + action = action[:punc+1] shown_action = "> You" + action console_print(second_to_first_person(shown_action)) story_manager.act(action) diff --git a/story/story_manager.py b/story/story_manager.py index ff0566f..d980467 100644 --- a/story/story_manager.py +++ b/story/story_manager.py @@ -67,11 +67,11 @@ class Story(): def latest_result(self): - latest_result = self.context mem_ind = self.memory if len(self.results) < 2: - latest_result += self.story_start - + latest_result = self.story_start + else: + latest_result = self.context while mem_ind > 0: if len(self.results) >= mem_ind: diff --git a/story/utils.py b/story/utils.py index 686e4f7..f0d1658 100644 --- a/story/utils.py +++ b/story/utils.py @@ -89,7 +89,7 @@ def split_first_sentence(text): def cut_trailing_action(text): lines = text.split("\n") last_line = lines[-1] - if "you ask." in last_line or "You ask." in last_line or "you say." in last_line or "You say." in last_line: + if "you ask" in last_line or "You ask" in last_line or "you say" in last_line or "You say" in last_line: text = "\n".join(lines[0:-1]) return text