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

use spk_embedding when sft

Browse Source
This commit is contained in:
lyuxiang.lx
2024-07-10 17:49:32 +08:00
parent a723ea375e
commit 0fd15bb12b
5 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
cosyvoice/utils/executor.py
View File

@@ -52,6 +52,10 @@ class Executor:
info_dict["batch_idx"] = batch_idx
if cosyvoice_join(group_join, info_dict):
break
if info_dict["use_spk_embedding"] is True:
batch_dict["embedding"] = batch_dict["spk_embedding"]
else:
batch_dict["embedding"] = batch_dict["utt_embedding"]
# Disable gradient synchronizations across DDP processes.
# Within this context, gradients will be accumulated on module