diff --git a/indextts/cli.py b/indextts/cli.py
index 5349901..39cd047 100644
--- a/indextts/cli.py
+++ b/indextts/cli.py
@@ -47,10 +47,10 @@ def main():
     if args.device is None:
         if torch.cuda.is_available():
             args.device = "cuda:0"
-        elif torch.mps.is_available():
-            args.device = "mps"
-        elif torch.xpu.is_available():
+        elif hasattr(torch, "xpu") and torch.xpu.is_available():
             args.device = "xpu"
+        elif hasattr(torch, "mps") and torch.mps.is_available():
+            args.device = "mps"
         else:
             args.device = "cpu"
             args.fp16 = False # Disable FP16 on CPU