[BUG] Always crashes after around the 15th reply about aidungeon HOT 24 OPEN

Traceback (most recent call last):
  File "/usr/lib/python3.7/site-packages/tensorflow_core/python/client/session.py", line 1365, in _do_call
    return fn(*args)
  File "/usr/lib/python3.7/site-packages/tensorflow_core/python/client/session.py", line 1350, in _run_fn
    target_list, run_metadata)
  File "/usr/lib/python3.7/site-packages/tensorflow_core/python/client/session.py", line 1443, in _call_tf_sessionrun
    run_metadata)
tensorflow.python.framework.errors_impl.InvalidArgumentError: indices[0,0] = 1024 is not in [0, 1024)
         [[{{node sample_sequence/while/model/GatherV2_1}}]]

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "play.py", line 211, in <module>
    play_aidungeon_2()
  File "play.py", line 180, in play_aidungeon_2
    result = "\n" + story_manager.act(action)
  File "/Fast-Games/AI Dungeon 2/story/story_manager.py", line 181, in act
    result = self.generate_result(action_choice)
  File "/Fast-Games/AI Dungeon 2/story/story_manager.py", line 186, in generate_result
    block = self.generator.generate(self.story_context() + action)
  File "/Fast-Games/AI Dungeon 2/generator/gpt2/gpt2_generator.py", line 108, in generate
    text = self.generate_raw(prompt)
  File "/Fast-Games/AI Dungeon 2/generator/gpt2/gpt2_generator.py", line 91, in generate_raw
    self.context: [context_tokens for _ in range(self.batch_size)]
  File "/usr/lib/python3.7/site-packages/tensorflow_core/python/client/session.py", line 956, in run
    run_metadata_ptr)
  File "/usr/lib/python3.7/site-packages/tensorflow_core/python/client/session.py", line 1180, in _run
    feed_dict_tensor, options, run_metadata)
  File "/usr/lib/python3.7/site-packages/tensorflow_core/python/client/session.py", line 1359, in _do_run
    run_metadata)
  File "/usr/lib/python3.7/site-packages/tensorflow_core/python/client/session.py", line 1384, in _do_call
    raise type(e)(node_def, op, message)
tensorflow.python.framework.errors_impl.InvalidArgumentError: indices[0,0] = 1024 is not in [0, 1024)
         [[node sample_sequence/while/model/GatherV2_1 (defined at /usr/lib/python3.7/site-packages/tensorflow_core/python/framework/ops.py:1748) ]]

Original stack trace for 'sample_sequence/while/model/GatherV2_1':
  File "play.py", line 211, in <module>
    play_aidungeon_2()
  File "play.py", line 74, in play_aidungeon_2
    generator = GPT2Generator()
  File "/Fast-Games/AI Dungeon 2/generator/gpt2/gpt2_generator.py", line 44, in __init__
    temperature=temperature, top_k=top_k, top_p=top_p
  File "/Fast-Games/AI Dungeon 2/generator/gpt2/src/sample.py", line 112, in sample_sequence
    back_prop=False,
  File "/usr/lib/python3.7/site-packages/tensorflow_core/python/ops/control_flow_ops.py", line 2753, in while_loop
    return_same_structure)
  File "/usr/lib/python3.7/site-packages/tensorflow_core/python/ops/control_flow_ops.py", line 2245, in BuildLoop
    pred, body, original_loop_vars, loop_vars, shape_invariants)
  File "/usr/lib/python3.7/site-packages/tensorflow_core/python/ops/control_flow_ops.py", line 2170, in _BuildLoop
    body_result = body(*packed_vars_for_body)
  File "/usr/lib/python3.7/site-packages/tensorflow_core/python/ops/control_flow_ops.py", line 2705, in <lambda>
    body = lambda i, lv: (i + 1, orig_body(*lv))
  File "/Fast-Games/AI Dungeon 2/generator/gpt2/src/sample.py", line 82, in body
    next_outputs = step(hparams, prev, past=past)
  File "/Fast-Games/AI Dungeon 2/generator/gpt2/src/sample.py", line 70, in step
    lm_output = model.model(hparams=hparams, X=tokens, past=past, reuse=tf.AUTO_REUSE)
  File "/Fast-Games/AI Dungeon 2/generator/gpt2/src/model.py", line 157, in model
    h = tf.gather(wte, X) + tf.gather(wpe, positions_for(X, past_length))
  File "/usr/lib/python3.7/site-packages/tensorflow_core/python/util/dispatch.py", line 180, in wrapper
    return target(*args, **kwargs)
  File "/usr/lib/python3.7/site-packages/tensorflow_core/python/ops/array_ops.py", line 3956, in gather
    params, indices, axis, name=name)
  File "/usr/lib/python3.7/site-packages/tensorflow_core/python/ops/gen_array_ops.py", line 4082, in gather_v2
    batch_dims=batch_dims, name=name)
  File "/usr/lib/python3.7/site-packages/tensorflow_core/python/framework/op_def_library.py", line 794, in _apply_op_helper
    op_def=op_def)
  File "/usr/lib/python3.7/site-packages/tensorflow_core/python/util/deprecation.py", line 507, in new_func
    return func(*args, **kwargs)
  File "/usr/lib/python3.7/site-packages/tensorflow_core/python/framework/ops.py", line 3357, in create_op
    attrs, op_def, compute_device)
  File "/usr/lib/python3.7/site-packages/tensorflow_core/python/framework/ops.py", line 3426, in _create_op_internal
    op_def=op_def)
  File "/usr/lib/python3.7/site-packages/tensorflow_core/python/framework/ops.py", line 1748, in __init__
    self._traceback = tf_stack.extract_stack()

AI Dungeon 2 will save and use your actions and game to continually improve AI
 Dungeon. If you would like to disable this enter 'nosaving' for any action.
 This will also turn off the ability to save games.

Initializing AI Dungeon! (This might take a few minutes)




 ▄▄▄       ██▓   ▓█████▄  █    ██  ███▄    █   ▄████ ▓█████  ▒█████   ███▄    █    
▒████▄    ▓██▒   ▒██▀ ██▌ ██  ▓██▒ ██ ▀█   █  ██▒ ▀█▒▓█   ▀ ▒██▒  ██▒ ██ ▀█   █    
▒██  ▀█▄  ▒██▒   ░██   █▌▓██  ▒██░▓██  ▀█ ██▒▒██░▄▄▄░▒███   ▒██░  ██▒▓██  ▀█ ██▒   
░██▄▄▄▄██ ░██░   ░▓█▄   ▌▓▓█  ░██░▓██▒  ▐▌██▒░▓█  ██▓▒▓█  ▄ ▒██   ██░▓██▒  ▐▌██▒   
 ▓█   ▓██▒░██░   ░▒████▓ ▒▒█████▓ ▒██░   ▓██░░▒▓███▀▒░▒████▒░ ████▓▒░▒██░   ▓██░   
 ▒▒   ▓▒█░░▓      ▒▒▓  ▒ ░▒▓▒ ▒ ▒ ░ ▒░   ▒ ▒  ░▒   ▒ ░░ ▒░ ░░ ▒░▒░▒░ ░ ▒░   ▒ ▒    
  ▒   ▒▒ ░ ▒ ░    ░ ▒  ▒ ░░▒░ ░ ░ ░ ░░   ░ ▒░  ░   ░  ░ ░  ░  ░ ▒ ▒░ ░ ░░   ░ ▒░   
  ░   ▒    ▒ ░    ░ ░  ░  ░░░ ░ ░    ░   ░ ░ ░ ░   ░    ░   ░ ░ ░ ▒     ░   ░ ░    
      ░  ░ ░        ░       ░              ░       ░    ░  ░    ░ ░           ░    
                  ░                                                                




Pick a setting.
0) fantasy (recommended)
1) mystery
2) apocalyptic
3) zombies
4) custom
Enter the number of your choice: 4

Enter a prompt that describes who you are and the first couple sentences of where
 you start out ex:
 'You are a knight in the kingdom of Larion. You are hunting the evil dragon
 who has been terrorizing the kingdom. You enter the forest searching for the
 dragon and see' 
Starting Prompt: nothingness

AI Dungeon 2 Instructions:
 Enter actions starting with a verb ex. "go to the tavern" or "attack the orc."
 To speak enter 'say "(thing you want to say)"' or just "(thing you want to say)"
 

The following commands can be entered for any action: 
  "revert"   Reverts the last action allowing you to pick a different action.
  "quit"     Quits the game and saves
  "restart"  Starts a new game and saves your current one
  "save"     Makes a new save of your game and gives you the save ID
  "load"     Asks for a save ID and loads the game if the ID is valid
  "print"    Prints a transcript of your adventure (without extra newline formatting)
  "help"     Prints these instructions again

Generating story...


nothingness.
> nothingness

You feel the air in your lungs begin to constrict, and you cough violently as
 you try to force the air out of your body. The pain is excruciating, but it's
 not enough to stop you from trying to breathe.

"What are you doing?" a voice says
> nothingness

The voice speaks again, this time with a cold tone. "Nothingness".
You don't know what that means, but you're going to find out soon.
 
> nothingness

You hear the sound of a door opening up behind you, and turn around to see a
 man standing there. He wears a black suit and has short brown hair. His eyes
 seem to be staring right through you, though. "I'm sorry about earlier," he
 begins
> nothingness

"No need for apologies", you reply. "It was an accident".
> nothingness

"Well, I suppose it was an accident. But I still think you should apologize".
> nothingness

"Nonsense! You can't just go apologizing to me all the time!"
> nothingness

"Okay, okay, I get it. I'll apologize to you later".
> nothingness

"Yeah, yeah, I get it. Goodbye".
> nothingness

With those words, the man turns away from you, leaving you alone in the room.
 You sit down on one of the chairs, feeling drained. You close your eyes, and
 fall asleep..
> nothingness

You wake up, lying on a bed. It looks like the same one you were sleeping in,
 except now it's covered in blood. There's also a knife stuck into your chest.

"Huh? What...what happened?" you ask.


> nothingness

You look at yourself in the mirror, which seems to have changed its appearance.
 Your face appears much older than when you first woke up. Your hair is also
 different, and now hangs down past your shoulders. Your skin is pale, and your
 teeth are white
> nothingness

Your head starts to hurt, and you start to shake. You scream out in pain, before
 falling back onto the mattress. A nurse comes running over, and takes off her
 gloves. She examines you, then leaves.

A few hours pass, and you realize that you haven't moved
> nothingness

Suddenly, you open your eyes. You're in a hospital room, with a woman sitting
 across from you. Her eyes are red, and she's crying. "I'm so sorry..". she says.
"Who are you?", you ask.

> nothingness

She doesn't answer, instead continuing to cry. "I'm so sorry..".
> nothingness

She continues to cry, and you take a deep breath. "Why did you do it?".
> nothingness

"Because I wanted to".
> nothingness

"Yes, because of my desire to kill you".
> nothingness
Woops that action caused the model to start looping. Try a different action to
 prevent that.
> nothingness

"So why didn't you tell anyone?"
> nothingness

"Because I thought no-one would believe me".
> nothingness

"How could you do such a thing?"
> you attempt to say something different for the 21st reply

"I don't understand".
> "you wouldn't understand"

>Check the closet.
You check the closet.
You check the closet.
You check the closet.
You check the closet.
You check the closet.
You check the closet.
You check the closet.
You check the closet.
You check the closet.

>back away
You check the.
You check the.
You check the.
You check the.
You check the.
You check the.
You check the.
You check the.
You check the.
You check the.
You check the.
You check the.