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

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

setup.py

@@ -27,6 +27,7 @@ requirements = {
         # ASR
         "sentencepiece",
         "jieba",
+        "rotary_embedding_torch",
         # TTS
         "pypinyin>=0.44.0",
         "espnet_tts_frontend",