Looks like an off-by-one error.
Python.Runtime.PythonException
HResult=0x80131500
Message=InvalidArgumentError : indices[0,0] = 1024 is not in [0, 1024)
[[Node: sample_sequence/while/model/GatherV2_1 = GatherV2[Taxis=DT_INT32, Tindices=DT_INT32, Tparams=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:CPU:0"](sample_sequence/while/model/GatherV2_1/Enter, sample_sequence/while/model/ExpandDims, sample_sequence/while/model/h11/attn/range_1/start, ^sample_sequence/while/strided_slice/stack_2)]]
Caused by op 'sample_sequence/while/model/GatherV2_1', defined at:
File "C:\Users<SCRAMBLED>\AppData\Local\conda\conda\envs\tf\lib\site-packages\tensorflow\python\ops\control_flow_ops.py", line 3232, in while_loop
return_same_structure)
File "C:\Users<SCRAMBLED>\AppData\Local\conda\conda\envs\tf\lib\site-packages\tensorflow\python\ops\control_flow_ops.py", line 2952, in BuildLoop
pred, body, original_loop_vars, loop_vars, shape_invariants)
File "C:\Users<SCRAMBLED>\AppData\Local\conda\conda\envs\tf\lib\site-packages\tensorflow\python\ops\control_flow_ops.py", line 2887, in _BuildLoop
body_result = body(*packed_vars_for_body)
File "C:\Users<SCRAMBLED>\AppData\Local\conda\conda\envs\tf\lib\site-packages\tensorflow\python\ops\control_flow_ops.py", line 3201, in
body = lambda i, lv: (i + 1, orig_body(*lv))
File "C:\Users<SCRAMBLED>\AppData\Local\conda\conda\envs\tf\lib\site-packages\tensorflow\python\ops\array_ops.py", line 2659, in gather
return gen_array_ops.gather_v2(params, indices, axis, name=name)
File "C:\Users<SCRAMBLED>\AppData\Local\conda\conda\envs\tf\lib\site-packages\tensorflow\python\ops\gen_array_ops.py", line 3761, in gather_v2
"GatherV2", params=params, indices=indices, axis=axis, name=name)
File "C:\Users<SCRAMBLED>\AppData\Local\conda\conda\envs\tf\lib\site-packages\tensorflow\python\framework\op_def_library.py", line 787, in _apply_op_helper
op_def=op_def)
File "C:\Users<SCRAMBLED>\AppData\Local\conda\conda\envs\tf\lib\site-packages\tensorflow\python\util\deprecation.py", line 454, in new_func
return func(*args, **kwargs)
File "C:\Users<SCRAMBLED>\AppData\Local\conda\conda\envs\tf\lib\site-packages\tensorflow\python\framework\ops.py", line 3155, in create_op
op_def=op_def)
File "C:\Users<SCRAMBLED>\AppData\Local\conda\conda\envs\tf\lib\site-packages\tensorflow\python\framework\ops.py", line 1717, in init
self._traceback = tf_stack.extract_stack()
InvalidArgumentError (see above for traceback): indices[0,0] = 1024 is not in [0, 1024)
[[Node: sample_sequence/while/model/GatherV2_1 = GatherV2[Taxis=DT_INT32, Tindices=DT_INT32, Tparams=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:CPU:0"](sample_sequence/while/model/GatherV2_1/Enter, sample_sequence/while/model/ExpandDims, sample_sequence/while/model/h11/attn/range_1/start, ^sample_sequence/while/strided_slice/stack_2)]]
Source=Python.Runtime
StackTrace:
[' File "C:\Users\\AppData\Local\conda\conda\envs\tf\lib\site-packages\tensorflow\python\client\session.py", line 877, in run\n run_metadata_ptr)\n', ' File "C:\Users\\AppData\Local\conda\conda\envs\tf\lib\site-packages\tensorflow\python\client\session.py", line 1100, in _run\n feed_dict_tensor, options, run_metadata)\n', ' File "C:\Users\\AppData\Local\conda\conda\envs\tf\lib\site-packages\tensorflow\python\client\session.py", line 1272, in _do_run\n run_metadata)\n', ' File "C:\Users\\AppData\Local\conda\conda\envs\tf\lib\site-packages\tensorflow\python\client\session.py", line 1291, in _do_call\n raise type(e)(node_def, op, message)\n']