fix(confucius): list supported device codes in the install hint

This commit is contained in:
Palash Debnath
2026-09-07 13:17:37 +05:30
parent 0921292319
commit 4008499ea5
2 changed files with 3 additions and 2 deletions
+1 -1
View File
@@ -2300,7 +2300,7 @@ _INSTALL_HINTS: dict[str, str] = {
"pockettts": "uv sync --extra pockettts (Kyutai, CPU-only, ~100 MB model on first use; MIT code + CC-BY-4.0 weights; HF-gated, review terms and set HF_TOKEN)",
"moss-tts-v15": "git clone OpenMOSS/MOSS-TTS + set OMNIVOICE_MOSS_TTS_V15_DIR (own venv, transformers==5.0; 8B, ~16 GB weights; CUDA/CPU, no MPS; Apache-2.0)",
"dots-tts": "git clone rednote-hilab/dots.tts + set OMNIVOICE_DOTS_TTS_DIR (own venv, transformers==4.57; 2B, ~9 GB weights; CUDA/CPU, Linux/macOS only — no Windows; Apache-2.0)",
"confucius4-tts":"git clone netease-youdao/Confucius4-TTS + set OMNIVOICE_CONFUCIUS4_TTS_DIR (own Python 3.10 venv; 14-lang cross-lingual zero-shot clone; ~5 GB weights auto-download; matching PyTorch accelerator runtime or CPU, no MPS; Apache-2.0)",
"confucius4-tts":"git clone netease-youdao/Confucius4-TTS + set OMNIVOICE_CONFUCIUS4_TTS_DIR (own Python 3.10 venv; 14-lang cross-lingual zero-shot clone; ~5 GB weights auto-download; CUDA/ROCm/XPU/NPU/CPU, no MPS; Apache-2.0)",
}
+2 -1
View File
@@ -50,4 +50,5 @@ def test_catalog_metadata_does_not_exclude_supported_accelerators():
assert "CUDA/CPU" not in Confucius4Backend.display_name
assert "CUDA/CPU" not in _INSTALL_HINTS[Confucius4Backend.id]
assert "matching" in _INSTALL_HINTS[Confucius4Backend.id]
for family in Confucius4Backend.gpu_compat:
assert family in _INSTALL_HINTS[Confucius4Backend.id].lower()