Sfoglia il codice sorgente

Merge pull request #1693 from whiteshirt0429/main

Fix CosyVoice3 config error
Xiang Lyu 4 mesi fa
parent
commit
bc34459bb8
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  1. 1 1
      cosyvoice/llm/llm.py

+ 1 - 1
cosyvoice/llm/llm.py

@@ -698,7 +698,7 @@ class CosyVoice3LM(Qwen2LM):
         lm_output, lm_output_mask = self.llm(lm_input, lm_input_len.to(device))
         logits = self.llm_decoder(lm_output)
         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}
 
     @torch.inference_mode()