first

speech_to_speech/arguments_classes/facebookmms_tts_arguments.py

@@ -0,0 +1,21 @@
+from dataclasses import dataclass, field
+
+
+@dataclass
+class FacebookMMSTTSHandlerArguments:
+    facebook_mms_model_name: str = field(
+        default="facebook/mms-tts-eng",
+        metadata={"help": "The model name to use. Default is 'facebook/mms-tts-eng'."},
+    )
+    tts_language: str = field(
+        default="en",
+        metadata={"help": "The language code for the TTS model. Default is 'en' for English."},
+    )
+    facebook_mms_device: str = field(
+        default="cuda",
+        metadata={"help": "The device to use for the TTS model. Default is 'cuda'."},
+    )
+    facebook_mms_torch_dtype: str = field(
+        default="float32",
+        metadata={"help": "The torch data type to use for the TTS model. Default is 'float32'."},
+    )