Function startup_event

examples/api/openai_api.py:63–91

Load ChatTTS model and default speaker embedding when the application starts

()

@app.on_event("startup")
async def startup_event():
    """Load ChatTTS model and default speaker embedding when the application starts"""
    # Initialize ChatTTS and async lock
    app.state.chat = ChatTTS.Chat(get_logger("ChatTTS"))
    app.state.model_lock = asyncio.Lock()  # Use async lock instead of thread lock

    # Register text normalizers
    app.state.chat.normalizer.register("en", normalizer_en_nemo_text())
    app.state.chat.normalizer.register("zh", normalizer_zh_tn())

    logger.info("Initializing ChatTTS...")
    if app.state.chat.load(source="huggingface"):
        logger.info("Model loaded successfully.")
    else:
        logger.error("Model loading failed, exiting application.")
        raise RuntimeError("Failed to load ChatTTS model")

    # Load default speaker embedding
    # Preload all supported speaker embeddings into memory at startup to avoid repeated loading during runtime
    app.state.spk_emb_map = {}
    for voice, spk_path in VOICE_MAP.items():
        if os.path.exists(spk_path):
            app.state.spk_emb_map[voice] = torch.load(
                spk_path, map_location=torch.device("cpu")
            )
            logger.info(f"Preloading speaker embedding: {voice} -> {spk_path}")
        else:
            logger.warning(f"Speaker embedding not found: {spk_path}, skipping preload")
    app.state.spk_emb = app.state.spk_emb_map.get("default")  # Default embedding


# Request parameter whitelist