add MPS support for Apple Silicon

config.json

@@ -8,6 +8,7 @@
   "model_type": "inf5",
   "remove_sil": true,
   "speed": 1.0,
+  "device": "",
   "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, device: str, ckpt_path: str = "checkpoints/model_best.pt", vocab_path: str = "checkpoints/vocab.txt", 
-                 speed: float = 1.0, remove_sil: bool = True, **kwargs):
+    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 = "", **kwargs):
         super().__init__(**kwargs)
         self.ckpt_path = ckpt_path
         self.vocab_path = vocab_path