File size: 1,454 Bytes
c72e80d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
from dataclasses import dataclass, field
from typing import Optional
@dataclass
class ModuleArguments:
device: Optional[str] = field(
default=None,
metadata={"help": "If specified, overrides the device for all handlers."},
)
mode: Optional[str] = field(
default="socket",
metadata={
"help": "The mode to run the pipeline in. Either 'local' or 'socket'. Default is 'socket'."
},
)
local_mac_optimal_settings: bool = field(
default=False,
metadata={
"help": "If specified, sets the optimal settings for Mac OS. Hence whisper-mlx, MLX LM and MeloTTS will be used."
},
)
stt: Optional[str] = field(
default="whisper",
metadata={
"help": "The STT to use. Either 'whisper', 'whisper-mlx', and 'paraformer'. Default is 'whisper'."
},
)
llm: Optional[str] = field(
default="transformers",
metadata={
"help": "The LLM to use. Either 'transformers' or 'mlx-lm'. Default is 'transformers'"
},
)
tts: Optional[str] = field(
default="parler",
metadata={
"help": "The TTS to use. Either 'parler', 'melo', or 'chatTTS'. Default is 'parler'"
},
)
log_level: str = field(
default="info",
metadata={
"help": "Provide logging level. Example --log_level debug, default=warning."
},
)
|