Răsfoiți Sursa

Merge pull request #745 from garywill/patch-2

Rename misspelled list_avaliable_spks() to list_available_spks()
Xiang Lyu 11 luni în urmă
părinte
comite
0f6ff298dd
3 a modificat fișierele cu 3 adăugiri și 3 ștergeri
  1. 1 1
      README.md
  2. 1 1
      cosyvoice/cli/cosyvoice.py
  3. 1 1
      webui.py

+ 1 - 1
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)

+ 1 - 1
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
 

+ 1 - 1
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()