Traceback (most recent call last):
  File "/root/.local/lib/python3.6/site-packages/discord/client.py", line 270, in _run_event
    await coro(*args, **kwargs)
  File "bot.py", line 543, in on_message
    await game.add_to_queue(msg.author, msg.content)
  File "bot.py", line 142, in add_to_queue
    await self.consume_queue()
  File "bot.py", line 111, in consume_queue
    pool, self.story_manager.act, f'\n> {to_calc[0]} {to_calc[1]}.\n'), self.timeout, loop=loop)
  File "/usr/lib/python3.6/asyncio/tasks.py", line 358, in wait_for
    return fut.result()
  File "/usr/lib/python3.6/concurrent/futures/thread.py", line 56, in run
    result = self.fn(*self.args, **self.kwargs)
  File "/AIDungeon/story/story_manager.py", line 215, in act
    result = self.generate_result(action_choice)
  File "/AIDungeon/story/story_manager.py", line 220, in generate_result
    block = self.generator.generate(self.story_context() + action)
  File "/AIDungeon/generator/gpt2/gpt2_generator.py", line 120, in generate
    text = self.generate_raw(prompt)
  File "/AIDungeon/generator/gpt2/gpt2_generator.py", line 103, in generate_raw
    self.context: [context_tokens for _ in range(self.batch_size)]
  File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/client/session.py", line 956, in run
    run_metadata_ptr)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/client/session.py", line 1180, in _run
    feed_dict_tensor, options, run_metadata)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/client/session.py", line 1359, in _do_run
    run_metadata)
  File "/usr/local/lib/python3.6/dist-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,1024] = 1024 is not in [0, 1024)
	 [[node sample_sequence/model/GatherV2_1 (defined at /usr/local/lib/python3.6/dist-packages/tensorflow_core/python/framework/ops.py:1748) ]]

Original stack trace for 'sample_sequence/model/GatherV2_1':
  File "/usr/lib/python3.6/threading.py", line 884, in _bootstrap
    self._bootstrap_inner()
  File "/usr/lib/python3.6/threading.py", line 916, in _bootstrap_inner
    self.run()
  File "/usr/lib/python3.6/threading.py", line 864, in run
    self._target(*self._args, **self._kwargs)
  File "/usr/lib/python3.6/concurrent/futures/thread.py", line 69, in _worker
    work_item.run()
  File "/usr/lib/python3.6/concurrent/futures/thread.py", line 56, in run
    result = self.fn(*self.args, **self.kwargs)
  File "bot.py", line 70, in create_story_manager
    generator = GPT2Generator(censor=False)
  File "/AIDungeon/generator/gpt2/gpt2_generator.py", line 52, in __init__
    top_p=top_p,
  File "/AIDungeon/generator/gpt2/src/sample.py", line 103, in sample_sequence
    past, prev, output = body(None, context, context)
  File "/AIDungeon/generator/gpt2/src/sample.py", line 89, in body
    next_outputs = step(hparams, prev, past=past)
  File "/AIDungeon/generator/gpt2/src/sample.py", line 75, in step
    hparams=hparams, X=tokens, past=past, reuse=tf.AUTO_REUSE
  File "/AIDungeon/generator/gpt2/src/model.py", line 186, in model
    h = tf.gather(wte, X) + tf.gather(wpe, positions_for(X, past_length))
  File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/util/dispatch.py", line 180, in wrapper
    return target(*args, **kwargs)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/ops/array_ops.py", line 3956, in gather
    params, indices, axis, name=name)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/ops/gen_array_ops.py", line 4082, in gather_v2
    batch_dims=batch_dims, name=name)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/framework/op_def_library.py", line 794, in _apply_op_helper
    op_def=op_def)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/util/deprecation.py", line 507, in new_func
    return func(*args, **kwargs)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/framework/ops.py", line 3357, in create_op
    attrs, op_def, compute_device)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/framework/ops.py", line 3426, in _create_op_internal
    op_def=op_def)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/framework/ops.py", line 1748, in __init__
    self._traceback = tf_stack.extract_stack()