mps support

config.py

@@ -17,7 +17,7 @@ exp_root = "logs"
 python_exec = sys.executable or "python"
 if torch.cuda.is_available():
     infer_device = "cuda"
-elif torch.mps.is_available():
+elif torch.backends.mps.is_available():
     infer_device = "mps"
 else:
     infer_device = "cpu"