This commit is contained in:
Nick Walton
2019-09-27 11:53:57 -06:00
parent ffec019ee6
commit 1141c523ef
3 changed files with 3 additions and 7 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("fallout")
prompt = get_story_start("forest")
story_manager = UnconstrainedStoryManager(generator)
story_manager.start_new_story(prompt)
+1 -1
View File
@@ -218,7 +218,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@@", "tl@@", ";@@",
'**', "http://@@"]
'**', "http://@@", "Redd@@", "UP@@"]
if num_new_lines > self.max_new_lines:
forbidden_tokens.append("\n")
+1 -5
View File
@@ -3,11 +3,7 @@ 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 "
fallout: "It's been 13 years since the nuclear bombs went off starting this nuclear apocalypse. You're getting low on food and need to go find some. You leave your shelter and see "
forest: "You wake up in an old dark forest with no memory of how you got there. You look around and see "
action_verbs:
classic: ["You attack", "You tell", "You use", "You go"]