deepgeek/CosyVoice
2
0
Fork 0
mirror of https://github.com/FunAudioLLM/CosyVoice.git synced 2026-02-05 09:59:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #745 from garywill/patch-2

Browse Source 
Rename misspelled list_avaliable_spks() to list_available_spks()
This commit is contained in:
Xiang Lyu
2024-12-29 17:20:17 +08:00
committed by GitHub
parent 877cf1c873 bd4be3fc05
commit 0f6ff298dd
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@@ -156,7 +156,7 @@ for i, j in enumerate(cosyvoice.inference_instruct2('收到好友从远方寄来
```python ```python
cosyvoice = CosyVoice('pretrained_models/CosyVoice-300M-SFT', load_jit=True, load_onnx=False, fp16=True) cosyvoice = CosyVoice('pretrained_models/CosyVoice-300M-SFT', load_jit=True, load_onnx=False, fp16=True)
# sft usage # sft usage
print(cosyvoice.list_avaliable_spks()) print(cosyvoice.list_available_spks())
# change stream=True for chunk stream inference # change stream=True for chunk stream inference
for i, j in enumerate(cosyvoice.inference_sft('你好，我是通义生成式语音大模型，请问有什么可以帮您的吗？', '中文女', stream=False)): for i, j in enumerate(cosyvoice.inference_sft('你好，我是通义生成式语音大模型，请问有什么可以帮您的吗？', '中文女', stream=False)):
torchaudio.save('sft_{}.wav'.format(i), j['tts_speech'], cosyvoice.sample_rate) torchaudio.save('sft_{}.wav'.format(i), j['tts_speech'], cosyvoice.sample_rate)

2
cosyvoice/cli/cosyvoice.py
View File

@@ -55,7 +55,7 @@ class CosyVoice:
self.model.load_onnx('{}/flow.decoder.estimator.fp32.onnx'.format(model_dir)) self.model.load_onnx('{}/flow.decoder.estimator.fp32.onnx'.format(model_dir))
del configs del configs
def list_avaliable_spks(self): def list_available_spks(self):
spks = list(self.frontend.spk2info.keys()) spks = list(self.frontend.spk2info.keys())
return spks return spks

2
webui.py
View File

@@ -182,7 +182,7 @@ if __name__ == '__main__':
help='local path or modelscope repo id') help='local path or modelscope repo id')
args = parser.parse_args() args = parser.parse_args()
cosyvoice = CosyVoice2(args.model_dir) if 'CosyVoice2' in args.model_dir else CosyVoice(args.model_dir) 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 prompt_sr = 16000
default_data = np.zeros(cosyvoice.sample_rate) default_data = np.zeros(cosyvoice.sample_rate)
main() main()