我正在尝试用真实数据集微调大模型，但是在训练一个epoch后就显示无法运行多头注意力机制，无法使用NPU继续运行我的代码，有相应的解决措施吗？下面附件是我的具体样例代码