fix(confucius): list supported device codes in the install hint
This commit is contained in:
@@ -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)",
|
||||
}
|
||||
|
||||
|
||||
|
||||
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user