RuntimeError: Attempted to call `variable.set_data(tensor)`, but `variable` and `tensor` have incompatible tensor type.

[Zekun525]

Traceback (most recent call last):
File "/home/user/lhh/Lenna-main/Lenna-main/chat.py", line 138, in
main(sys.argv[1:])
File "/home/user/lhh/Lenna-main/Lenna-main/chat.py", line 51, in main
param[1].data = trained_weight[save_name]
RuntimeError: Attempted to call variable.set_data(tensor), but variable and tensor have incompatible tensor type.