第三阶段和第四阶段均有错误

[haizeiwanglf]

Describe the bug

Please provide a clear and concise description of what the bug is. If applicable, add screenshots to help explain your problem, especially for visualization related problems.
第三阶段提示model和input不在一个设备上
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cuda:0! (when checking argument for argument weight in method wrapper_CUDA__native_layer_norm)
第四阶段报错KeyError: 'instruction'