diff --git a/generator/gpt2/gpt2_generator.py b/generator/gpt2/gpt2_generator.py index 1bb7524..6de0e44 100644 --- a/generator/gpt2/gpt2_generator.py +++ b/generator/gpt2/gpt2_generator.py @@ -10,7 +10,7 @@ import numpy as np class GPT2Generator: - def __init__(self, generate_num=48, temperature=0.4, top_k=40, top_p=0.9): + def __init__(self, generate_num=42, temperature=0.3, top_k=40, top_p=0.9): self.generate_num=generate_num self.temp = temperature self.top_k = top_k diff --git a/generator/gpt2/src/sample.py b/generator/gpt2/src/sample.py index 5879d11..1efc334 100644 --- a/generator/gpt2/src/sample.py +++ b/generator/gpt2/src/sample.py @@ -17,7 +17,7 @@ def penalize_used(logits, output): return tf.compat.v1.where( bool_tensor, - logits / 1.2, + logits / 1.15, logits) diff --git a/play.py b/play.py index 6660d15..0723f94 100644 --- a/play.py +++ b/play.py @@ -42,9 +42,9 @@ def select_game(): def instructions(): text = "AI Dungeon 2 Instructions:" text += '\n* Enter actions starting with a verb ex. "go to the place" or "attack the orc"' - text += '\n* Just press enter with no action if you want to let the story continue.' - text += '\n* Enter "restart" for any action to end your game and start a new one' - text += '\n* Enjoy!' + text += '\n* If you want to say something then enter \'say "(thing you want to say)"\'' + text += '\n* You can also just press enter with no action if you want to let the story continue.' + text += '\n* Finally if you want to end your game and start a new one just enter "restart" for any action. ' return text def play_aidungeon_2(): diff --git a/story/story_data.yaml b/story/story_data.yaml index 298ac6e..0a84004 100644 --- a/story/story_data.yaml +++ b/story/story_data.yaml @@ -1,9 +1,9 @@ prompts: classic: "You enter a dungeon with your trusty sword and shield. You are searching for the evil necromancer who killed your family. You've heard that he resides at the bottom of the dungeon, guarded by legions of the undead. You enter the first door and see " - hospital: "You wake up in an old rundown hospital with no memory of how you got there. You look around and see " + hospital: "You wake up in an old rundown hospital with no memory of how you got there. You take a look around the room. " - peasant: "You were kicked out of your home and have set out to see if you can find work in the next town. You step onto the road and see " + peasant: "You were kicked out of your home and have set out to see if you can find work in the next town. You step onto the road and look around you. " 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 "