umbragit/Qwen3-TTS
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix finetuning bug

Browse Source
This commit is contained in:
Xiong Wang
2026-03-17 14:38:41 +08:00
committed by GitHub
parent 6cafe5582c
commit 022e286b98
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
finetuning/sft_12hz.py
View File

@@ -105,7 +105,7 @@ def train():
) )
hidden_states = outputs.hidden_states[0][-1] hidden_states = outputs.hidden_states[0][-1]
talker_hidden_states = hidden_states[codec_mask[:, 1:]] talker_hidden_states = hidden_states[codec_mask[:, :-1]]
talker_codec_ids = codec_ids[codec_mask] talker_codec_ids = codec_ids[codec_mask]
sub_talker_logits, sub_talker_loss = model.talker.forward_sub_talker_finetune(talker_codec_ids, talker_hidden_states) sub_talker_logits, sub_talker_loss = model.talker.forward_sub_talker_finetune(talker_codec_ids, talker_hidden_states)