Can support MPS device type?

#48
by LouiSum - opened

The source code seems to say only CUDA or CPU is supported, but MPS. How can I move the model to GPU on my Mac?

I also don't see an option, pow.Scaler_out is implemented in https://github.com/pytorch/pytorch/issues/77764 but I still get the error that mps is not enabled and comment on this given issue

Sign up or log in to comment