22 lines
751 B
Python
22 lines
751 B
Python
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'."},
|
|
)
|