torch.device Metal Performance Shaders error

pkhungurn / talking-head-anime-demo

Demo for the "Talking Head Anime from a Single Image."

MIT License

1.99k stars 286 forks source link

torch.device Metal Performance Shaders error #44

Closed JojiTokitou closed 11 months ago

JojiTokitou commented 1 year ago

I'm a Japanese student. Sorry for the poor English. When I tried this program, got error as this issue's title. I use M1 MacBook Air and GPU is Metal Performance Shaders (MPS). Can anybody offer me help？

JojiTokitou commented 1 year ago

I found this error

RuntimeError: don't know how to restore data location of torch.FloatStorage (tagged with meta)

dragonmeteor commented 1 year ago

I'm sorry to say that I might not be able to help with your problem. I don't have a MacBook with an M1 chip, so I cannot reproduce your problem.

Someone I know tried to make my system work on MacBooks. It seems that the PyTorch implementation for the M1 chip still have bugs and unimplemented features. So, you either can hack PyTorch to fix those bugs and insert unimplemented features or you can wait until the bugs are fixed by the PyTorch team.