12345678910111213141516171819202122232425262728293031323334353637383940414243444546 |
- 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', 'faster-whisper', 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', 'chatTTS' or 'facebookMMS'. Default is 'parler'"
- },
- )
- log_level: str = field(
- default="info",
- metadata={
- "help": "Provide logging level. Example --log_level debug, default=info."
- },
- )
|