add MPS support for Apple Silicon

config.json

@@ -8,7 +8,6 @@
   "model_type": "inf5",
   "remove_sil": true,
   "speed": 1.0,
-  "device": "cpu",
   "torch_dtype": "float32",
   "transformers_version": "4.49.0",
   "vocab_path": "checkpoints/vocab.txt"

model.py

@@ -24,8 +24,8 @@ import os
 class INF5Config(PretrainedConfig):
     model_type = "inf5"
 
-    def __init__(self, ckpt_path: str = "checkpoints/model_best.pt", vocab_path: str = "checkpoints/vocab.txt", 
-                 speed: float = 1.0, remove_sil: bool = True, device: str = "cpu", **kwargs):
+    def __init__(self, device: str, ckpt_path: str = "checkpoints/model_best.pt", vocab_path: str = "checkpoints/vocab.txt", 
+                 speed: float = 1.0, remove_sil: bool = True, **kwargs):
         super().__init__(**kwargs)
         self.ckpt_path = ckpt_path
         self.vocab_path = vocab_path
@@ -38,10 +38,7 @@ class INF5Model(PreTrainedModel):
 
     def __init__(self, config):
         super().__init__(config)
-        if config.device:
-            self._device = config.device
-        else:
-            self._device = "cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"
+        self._device = config.device if config.device else "cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"
 
         # Load vocoder
         self.vocoder = torch.compile(load_vocoder(vocoder_name="vocos", is_local=False, device=self._device))