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