我把paddlehub升级到了2.1.0也没用。。

应该是同样问题吧,期待能有解答吧，运行到了：loss, _, __ = model(
attn_ids,
sent_ids=tgt_sids,
pos_ids=tgt_pids,
attn_bias=mask_attn_2_srctgtattn,
past_cache=(past_cache_k, past_cache_v),
tgt_labels=tgt_labels,
tgt_pos=paddle.nonzero(attn_ids == attn_id))
报了错：
Traceback (most recent call last):
File "J:/***/ERNIE-GEN/erniePoetry.py", line 289, in
loss, _, __ = model(
File "D:\ProgramData\Anaconda3\lib\site-packages\paddle\fluid\dygraph\layers.py", line 902, in __call__
outputs = self.forward(*inputs, **kwargs)
File "D:\ProgramData\Anaconda3\lib\site-packages\paddlenlp\transformers\ernie_gen\modeling.py", line 610, in forward
loss = F.cross_entropy(
File "D:\ProgramData\Anaconda3\lib\site-packages\paddle\nn\functional\loss.py", line 1389, in cross_entropy
_, out = core.ops.softmax_with_cross_entropy(
ValueError: (InvalidArgument) Tensor holds the wrong type, it holds int, but desires to be int64_t.
[Hint: Expected valid == true, but received valid:0 != true:1.] (at C:\home\workspace\Paddle_release\paddle/fluid/framework/tensor_impl.h:33)
[operator < softmax_with_cross_entropy > error]

源代码来自项目：使用PaddleNLP预训练模型ERNIE-GEN生成诗歌
https://aistudio.baidu.com/aistudio/projectdetail/1339888?channelType=0&channel=0
除了注释掉了paddle.set_device('gpu')，都是原代码了。