This commit is contained in:
lyuxiang.lx
2024-12-13 15:33:36 +08:00
parent 9b8f28aa32
commit 6b9cebea14
3 changed files with 4 additions and 3 deletions

View File

@@ -128,7 +128,7 @@ class CosyVoice2(CosyVoice):
if not os.path.exists(model_dir):
model_dir = snapshot_download(model_dir)
with open('{}/cosyvoice.yaml'.format(model_dir), 'r') as f:
configs = load_hyperpyyaml(f)
configs = load_hyperpyyaml(f, overrides={'qwen_pretrain_path': os.path.join(model_dir, 'Qwen2-0.5B-CosyVoice-BlankEN')})
self.frontend = CosyVoiceFrontEnd(configs['get_tokenizer'],
configs['feat_extractor'],
'{}/campplus.onnx'.format(model_dir),