weixin_41670367的博客

原创 MAC M1 芯片训练huggingface上的模型，报错RuntimeError: Placeholder storage has not been allocated on MPS device!

后来看参数初始化的源码，发现在设置只使用mps的时候会有问题，在判断use_mps_device后device设置为“mps”，但是在use_cpu的判断里会在else里面判断是否能使用cuda，如果不能就设置为cpu，在这个判断之后device设置为“mps”，后续代码就会根据这个设置把input放到cpu中，就会报错。设置完.to(device)之后可以正常运行，注意tensors的格式需要是“pt”根据我的情况修改为elif就不会进最后的else语句里面了。