mirror of
https://github.com/FunAudioLLM/CosyVoice.git
synced 2026-02-04 17:39:25 +08:00
rename list_avaliable_spks() to list_available_spks()
Signed-off-by: garywill <garywill@disroot.org>
This commit is contained in:
@@ -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)
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
2
webui.py
2
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()
|
||||
|
||||
Reference in New Issue
Block a user