|
|
@@ -27,7 +27,7 @@ from pydantic import BaseModel, field_validator
|
|
|
# 配置
|
|
|
# ============================================================
|
|
|
|
|
|
-DEFAULT_MODEL_NAME = os.getenv("TTS_ONNX_MODEL_NAME", "model.onnx")
|
|
|
+DEFAULT_MODEL_NAME = os.getenv("TTS_ONNX_MODEL_NAME", "model_quantized.onnx")
|
|
|
MODEL_DIR = os.getenv("TTS_ONNX_MODEL_DIR", "/home/tts-server/onnx")
|
|
|
HF_MODEL_ID = os.getenv("TTS_ONNX_HF_MODEL_ID", "onnx-community/Kokoro-82M-ONNX")
|
|
|
TOKENIZER_PATH = os.getenv("TTS_ONNX_TOKENIZER_PATH", str(Path(MODEL_DIR) / "tokenizer.json"))
|
|
|
@@ -45,6 +45,14 @@ VOICE_ALIASES = {}
|
|
|
logging.basicConfig(level=getattr(logging, LOG_LEVEL, logging.INFO))
|
|
|
logger = logging.getLogger("speech_tts_onnx")
|
|
|
|
|
|
+
|
|
|
+class _PhonemizerWordCountMismatchFilter(logging.Filter):
|
|
|
+ def filter(self, record: logging.LogRecord) -> bool:
|
|
|
+ return "words count mismatch" not in record.getMessage()
|
|
|
+
|
|
|
+
|
|
|
+logging.getLogger("phonemizer").addFilter(_PhonemizerWordCountMismatchFilter())
|
|
|
+
|
|
|
Path(CACHE_DIR).mkdir(parents=True, exist_ok=True)
|
|
|
SENT_SPLIT_RE = re.compile(r"(?<=[.!?,:])\s+")
|
|
|
_TOKENIZER_CACHE = None
|
