This commit is contained in:
Nick
2019-09-26 20:06:52 -06:00
parent eaee91a93a
commit 97c895f8c3
3 changed files with 4 additions and 2 deletions
+1 -1
View File
@@ -23,7 +23,7 @@ def console_print(str, pycharm=False):
def play_unconstrained():
generator = CTRLGenerator()
#generator = WebGenerator(CRED_FILE)
prompt = get_story_start("hospital")
prompt = get_story_start("peasant")
story_manager = UnconstrainedStoryManager(generator)
story_manager.start_new_story(prompt)
+1 -1
View File
@@ -217,7 +217,7 @@ class CTRLGenerator():
# disallow some tokens
forbidden_tokens = ['<unk>', 'Sco@@', "&amp@@", "1]@@", "2]@@", "3]@@", "4]@@", "https://www.@@", "[@@", ":@@",
"Edit", "&@@", "2:","1:", ":", "Edit@@", "EDI@@", "EDIT@@", "edit", "TL@@"]
"Edit", "&@@", "2:","1:", ":", "Edit@@", "EDI@@", "EDIT@@", "edit", "TL@@", "tl@@", ";@@", '**']
if num_new_lines > self.max_new_lines:
forbidden_tokens.append("\n")
+2
View File
@@ -3,6 +3,8 @@ prompts:
hospital: "You wake up in an old rundown hospital with no memory of how you got there. You look around and see "
peasant: "You are a medieval peasant. You just finished the days work in the wheat fields. You get home to your thatched hut and see "
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 "
action_verbs: