From 6113567e94a885193f451d3d81665a609e9efbe5 Mon Sep 17 00:00:00 2001 From: Arcitec <38923130+Arcitec@users.noreply.github.com> Date: Wed, 10 Sep 2025 20:04:18 +0200 Subject: [PATCH] fix(cli): More robust device priority checks --- indextts/cli.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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