From bd4be3fc05437f190181a73a6efd3fd010b918dd Mon Sep 17 00:00:00 2001 From: garywill Date: Wed, 18 Dec 2024 11:19:43 +0800 Subject: [PATCH] rename list_avaliable_spks() to list_available_spks() Signed-off-by: garywill --- README.md | 2 +- cosyvoice/cli/cosyvoice.py | 2 +- webui.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 14347dc..0bd308f 100644 --- a/README.md +++ b/README.md @@ -156,7 +156,7 @@ for i, j in enumerate(cosyvoice.inference_instruct2('收到好友从远方寄来 ```python cosyvoice = CosyVoice('pretrained_models/CosyVoice-300M-SFT', load_jit=True, load_onnx=False, fp16=True) # sft usage -print(cosyvoice.list_avaliable_spks()) +print(cosyvoice.list_available_spks()) # change stream=True for chunk stream inference for i, j in enumerate(cosyvoice.inference_sft('你好,我是通义生成式语音大模型,请问有什么可以帮您的吗?', '中文女', stream=False)): torchaudio.save('sft_{}.wav'.format(i), j['tts_speech'], cosyvoice.sample_rate) diff --git a/cosyvoice/cli/cosyvoice.py b/cosyvoice/cli/cosyvoice.py index bd2a520..3f848a8 100644 --- a/cosyvoice/cli/cosyvoice.py +++ b/cosyvoice/cli/cosyvoice.py @@ -55,7 +55,7 @@ class CosyVoice: self.model.load_onnx('{}/flow.decoder.estimator.fp32.onnx'.format(model_dir)) del configs - def list_avaliable_spks(self): + def list_available_spks(self): spks = list(self.frontend.spk2info.keys()) return spks diff --git a/webui.py b/webui.py index 196718d..403b016 100644 --- a/webui.py +++ b/webui.py @@ -182,7 +182,7 @@ if __name__ == '__main__': help='local path or modelscope repo id') args = parser.parse_args() cosyvoice = CosyVoice2(args.model_dir) if 'CosyVoice2' in args.model_dir else CosyVoice(args.model_dir) - sft_spk = cosyvoice.list_avaliable_spks() + sft_spk = cosyvoice.list_available_spks() prompt_sr = 16000 default_data = np.zeros(cosyvoice.sample_rate) main()