gpu-tts-example/config.py at main · gpu-cli/gpu-tts-example · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
"""Configuration for TTS/STT services."""

import os

# --- TTS Config ---
TTS_MODEL: str = os.getenv("TTS_MODEL", "chatterbox")  # "chatterbox" or "chatterbox-turbo"
TTS_EXAGGERATION: float = float(os.getenv("TTS_EXAGGERATION", "0.5"))  # 0=monotone, 1=expressive
TTS_CFG_WEIGHT: float = float(os.getenv("TTS_CFG_WEIGHT", "0.5"))  # voice matching strength
TTS_DEVICE: str = os.getenv("TTS_DEVICE", "cuda")

# --- STT Config ---
STT_MODEL: str = os.getenv("STT_MODEL", "large-v3-turbo")  # tiny|base|small|medium|large-v3|large-v3-turbo
STT_COMPUTE_TYPE: str = os.getenv("STT_COMPUTE_TYPE", "float16")  # float16|int8|float32
STT_DEVICE: str = os.getenv("STT_DEVICE", "cuda")
STT_BEAM_SIZE: int = int(os.getenv("STT_BEAM_SIZE", "5"))

# --- Server Config ---
HOST: str = os.getenv("HOST", "0.0.0.0")
PORT: int = int(os.getenv("PORT", "8000"))