[BUG] finetune/dataset.py | TypeError：无法根据规则“same_kind”将数组数据从 dtype('float64') 转换为 dtype('int32')

[rover5056]

是否已有关于该错误的issue或讨论？ | Is there an existing issue / discussion for this?

• [X] 我已经搜索过已有的issues和讨论 | I have searched the existing issues / discussions

该问题是否在FAQ中有解答？ | Is there an existing answer for this in FAQ?

• [X] 我已经搜索过FAQ | I have searched FAQ

当前行为 | Current Behavior

dataset.py

ids = torch.from_numpy(np.hstack(input_ids, dtype=np.int32)) 这里直接转换类型会报错

手动在 input_ids.append(prefix_ids) 之前把 prefix_ids 转一下就可以。。 prefix_ids = np.array(prefix_ids,dtype=np.int32) message_ids = np.array(message_ids,dtype=np.int32) 辛苦看看是不是 bug

期望行为 | Expected Behavior

No response

复现方法 | Steps To Reproduce

No response

运行环境 | Environment

- OS:
- Python:
- Transformers:
- PyTorch:
- CUDA (`python -c 'import torch; print(torch.version.cuda)'`):

备注 | Anything else?

No response

[YuzaChongyi]

之前有一个类似的 https://github.com/OpenBMB/MiniCPM-V/issues/113, 是训练数据出现了空消息，这个时候 input_ids, 中会包含一个 [] 导致该报错，你可以检查一下训练数据是否出现了空的字段。