Merge pull request #728 from FunAudioLLM/dev/lyuxiang.lx

update
This commit is contained in:
Xiang Lyu
2024-12-16 15:34:26 +08:00
committed by GitHub
3 changed files with 4 additions and 3 deletions

View File

@@ -142,7 +142,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, overrides={'qwen_pretrain_path': os.path.join(model_dir, 'Qwen2-0.5B-CosyVoice-BlankEN')})
configs = load_hyperpyyaml(f, overrides={'qwen_pretrain_path': os.path.join(model_dir, 'CosyVoice-BlankEN')})
self.frontend = CosyVoiceFrontEnd(configs['get_tokenizer'],
configs['feat_extractor'],
'{}/campplus.onnx'.format(model_dir),