Merge pull request adefossez#1 from elidourado/patch-2

Use mps by default if available
jjreisfl · Dec 8, 2023 · 08cf353 · 08cf353
2 parents f774e03 + 6a5cefa
commit 08cf353
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/demucs/separate.py b/demucs/separate.py
@@ -41,7 +41,13 @@ def get_parser():
                         'Default is "{track}/{stem}.{ext}".')
     parser.add_argument("-d",
                         "--device",
-                        default="cuda" if th.cuda.is_available() else "cpu",
+                        default=(
+                            "cuda"
+                            if th.cuda.is_available()
+                            else "mps"
+                            if th.backends.mps.is_available()
+                            else "cpu"
+                        ),
                         help="Device to use, default is cuda if available else cpu")
     parser.add_argument("--shifts",
                         default=1,