在117 M预训练模型上运行条件文本生成脚本时，对于输入文本有点大，有什么想法吗？
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/start, ^sample_sequence/while/model/h11/attn/strided_slice_8/stack_2)]]