supertonic.config¶

logger = getLogger(__name__)

DEFAULT_MODEL_REPO = getenv(
    "SUPERTONIC_MODEL_REPO", "Supertone/supertonic"
)

DEFAULT_CACHE_DIR = getenv(
    "SUPERTONIC_CACHE_DIR",
    str(home() / ".cache" / "supertonic"),
)

DEFAULT_MODEL_REVISION = getenv(
    "SUPERTONIC_MODEL_REVISION", "v1.0.0"
)

ONNX_DIR = Path('onnx')

VOICE_STYLES_DIR = Path('voice_styles')

CFG_REL_PATH = ONNX_DIR / 'tts.json'

UNICODE_INDEXER_REL_PATH = ONNX_DIR / "unicode_indexer.json"

DP_ONNX_REL_PATH = ONNX_DIR / 'duration_predictor.onnx'

TEXT_ENC_ONNX_REL_PATH = ONNX_DIR / 'text_encoder.onnx'

VECTOR_EST_ONNX_REL_PATH = (
    ONNX_DIR / "vector_estimator.onnx"
)

VOCODER_ONNX_REL_PATH = ONNX_DIR / 'vocoder.onnx'

DEFAULT_TOTAL_STEPS = 5

DEFAULT_SPEED = 1.05

DEFAULT_MAX_CHUNK_LENGTH = 300

DEFAULT_SILENCE_DURATION = 0.3

MIN_SPEED = 0.7

MAX_SPEED = 2.0

MIN_TOTAL_STEPS = 1

MAX_TOTAL_STEPS = 100

DEFAULT_ONNX_PROVIDERS = ['CPUExecutionProvider']

DEFAULT_INTRA_OP_NUM_THREADS = _parse_env_int(
    "SUPERTONIC_INTRA_OP_THREADS"
)

DEFAULT_INTER_OP_NUM_THREADS = _parse_env_int(
    "SUPERTONIC_INTER_OP_THREADS"
)

MAX_TEXT_LENGTH = 100000

LOG_FORMAT = (
    "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
)

LOG_LEVEL = getenv('SUPERTONIC_LOG_LEVEL', 'INFO')