update

X-LANCE · Aug 10, 2024 · 67efd2c · 67efd2c
1 parent 8a51ce4
commit 67efd2c
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/code/demo.py b/code/demo.py
@@ -294,11 +294,8 @@ def main(args):
 
     # macOS Config
     # Check if MPS is available
-    if torch.backends.mps.is_available():
+    if hasattr(torch.backends, 'mps') and torch.backends.mps.is_available():
         args.device = torch.device("mps")
         print("MPS backend is available.")
-    # else:
-    #     args.device = torch.device("cpu")
-    #     print("MPS backend is not available. Using CPU instead.")
 
     main(args)