- 博客(1)
- 收藏
- 关注
原创 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语句里面了。
2024-07-25 11:23:18
373
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人