Importing torch does not import torch.mps so the call failed.
Attach a file by drag & drop or click to upload