fix(cli): More robust device priority checks

2025-09-10 20:04:18 +02:00 · 2025-09-10 20:04:18 +02:00 · 6113567e94
commit 6113567e94
parent c3d7ab4adc
1 changed files with 3 additions and 3 deletions
--- 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