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

Merge pull request #1693 from whiteshirt0429/main

Browse Source 
Fix CosyVoice3 config error
This commit is contained in:
Xiang Lyu
2025-12-22 13:56:19 +08:00
committed by GitHub
parent 9f27b42cd9 ff0d05c380
commit bc34459bb8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cosyvoice/llm/llm.py
View File

@@ -698,7 +698,7 @@ class CosyVoice3LM(Qwen2LM):
lm_output, lm_output_mask = self.llm(lm_input, lm_input_len.to(device)) lm_output, lm_output_mask = self.llm(lm_input, lm_input_len.to(device))
logits = self.llm_decoder(lm_output) logits = self.llm_decoder(lm_output)
loss = self.criterion_ce(logits, lm_target.to(device)) loss = self.criterion_ce(logits, lm_target.to(device))
acc = th_accuracy(logits.view(-1, self.speech_token_size + 3), lm_target, ignore_label=IGNORE_ID) acc = th_accuracy(logits.view(-1, self.speech_token_size + 200), lm_target, ignore_label=IGNORE_ID)
return {'loss': loss, 'acc': acc} return {'loss': loss, 'acc': acc}
@torch.inference_mode() @torch.inference_mode()