dekun
eb71e28427
Generate ChatTTS sample_random_speaker presets without cloud APIs; choose clone or preset in synthesize UI. Co-authored-by: Cursor <cursoragent@cursor.com>
81 lines
2.1 KiB
Python
81 lines
2.1 KiB
Python
#!/usr/bin/env python3
|
|
"""生成 ChatTTS 本地预设说话人（sample_random_speaker，走 GPU）。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import sys
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
|
|
ROOT = Path(__file__).resolve().parent.parent
|
|
sys.path.insert(0, str(ROOT))
|
|
|
|
import torch
|
|
|
|
from config import CHATTTS_MODEL_DIR
|
|
from tts_service import get_chattts_instance, reset_chattts_instance
|
|
from voice_presets import (
|
|
DEFAULT_MANIFEST,
|
|
MANIFEST_PATH,
|
|
PRESETS_DIR,
|
|
VOICES_DIR,
|
|
ensure_manifest,
|
|
)
|
|
|
|
|
|
def main() -> None:
|
|
ensure_manifest()
|
|
PRESETS_DIR.mkdir(parents=True, exist_ok=True)
|
|
|
|
try:
|
|
from whisper_service import reset_whisper_model
|
|
|
|
reset_whisper_model()
|
|
except Exception:
|
|
pass
|
|
|
|
reset_chattts_instance()
|
|
chat, err = get_chattts_instance()
|
|
if chat is None:
|
|
raise SystemExit(f"ChatTTS 加载失败: {err}")
|
|
|
|
if not hasattr(chat, "sample_random_speaker"):
|
|
raise SystemExit("当前 ChatTTS 版本不支持 sample_random_speaker")
|
|
|
|
presets = DEFAULT_MANIFEST["presets"]
|
|
print(f"[INFO] 生成 {len(presets)} 个预设音色 → {PRESETS_DIR}")
|
|
|
|
for item in presets:
|
|
pid = item["id"]
|
|
label = item["label"]
|
|
out_path = PRESETS_DIR / f"{pid}.pt"
|
|
|
|
spk_emb = chat.sample_random_speaker()
|
|
payload = {
|
|
"version": 1,
|
|
"preset": True,
|
|
"id": pid,
|
|
"label": label,
|
|
"spk_emb": spk_emb,
|
|
"spk_smp": None,
|
|
"txt_smp": "",
|
|
"created_at": datetime.now().isoformat(),
|
|
"source": "ChatTTS.sample_random_speaker",
|
|
}
|
|
torch.save(payload, out_path)
|
|
print(f" [OK] {label} → {out_path.name}")
|
|
|
|
manifest = json.loads(MANIFEST_PATH.read_text(encoding="utf-8"))
|
|
manifest["generated_at"] = datetime.now().isoformat()
|
|
manifest["chattts_model"] = str(CHATTTS_MODEL_DIR)
|
|
MANIFEST_PATH.write_text(
|
|
json.dumps(manifest, ensure_ascii=False, indent=2),
|
|
encoding="utf-8",
|
|
)
|
|
print("[OK] 全部预设音色生成完成")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|