fix(gpt): device parsing

2noise · Oct 10, 2024 · 57f2cc8 · 57f2cc8
1 parent b3a180e
commit 57f2cc8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ChatTTS/utils/gpu.py b/ChatTTS/utils/gpu.py
@@ -30,7 +30,7 @@ def select_device(min_memory=2047, experimental=False):
             )
             device = torch.device("cpu")
         else:
-            device = provider.device(dev_idx)
+            device = provider._get_device(dev_idx)
     elif torch.backends.mps.is_available():
         """
         Currently MPS is slower than CPU while needs more memory and core utility,