Merge branch 'main' of https://github.com/alibaba-damo-academy/FunASR into main

2025-09-15 14:48:36 +08:00 · 2023-08-11 10:56:38 +08:00 · 2023-08-11 10:56:38 +08:00 · f5da7be222
commit f5da7be222
parent 44e7762ce6 ecccb37896
2 changed files with 3 additions and 2 deletions
--- a/funasr/models/e2e_asr_transducer.py
+++ b/funasr/models/e2e_asr_transducer.py
@ -697,7 +697,7 @@ class UnifiedTransducerModel(FunASRModel):

        loss_trans = loss_trans_utt + loss_trans_chunk
        loss_ctc = loss_ctc + loss_ctc_chunk
-        loss_ctc = loss_att + loss_att_chunk
+        loss_att = loss_att + loss_att_chunk

        loss = (
            self.transducer_weight * loss_trans
@ -990,4 +990,4 @@ class UnifiedTransducerModel(FunASRModel):
            ignore_label=self.ignore_id,
        )

-        return loss_att, acc_att
+        return loss_att, acc_att
--- a/setup.py
+++ b/setup.py
@ -27,6 +27,7 @@ requirements = {
        # ASR
        "sentencepiece",
        "jieba",
+        "rotary_embedding_torch",
        # TTS
        "pypinyin>=0.44.0",
        "espnet_tts_frontend",