嘉渊 vor 2 Jahren
Ursprung
Commit
c439eed31f
2 geänderte Dateien mit 3 neuen und 3 gelöschten Zeilen
  1. 2 2
      funasr/models/e2e_asr_contextual_paraformer.py
  2. 1 1
      funasr/models/e2e_asr_mfcca.py

+ 2 - 2
funasr/models/e2e_asr_contextual_paraformer.py

@@ -43,9 +43,7 @@ class NeatContextualParaformer(Paraformer):
         frontend: Optional[AbsFrontend],
         specaug: Optional[AbsSpecAug],
         normalize: Optional[AbsNormalize],
-        preencoder: Optional[AbsPreEncoder],
         encoder: AbsEncoder,
-        postencoder: Optional[AbsPostEncoder],
         decoder: AbsDecoder,
         ctc: CTC,
         ctc_weight: float = 0.5,
@@ -72,6 +70,8 @@ class NeatContextualParaformer(Paraformer):
         crit_attn_weight: float = 0.0,
         crit_attn_smooth: float = 0.0,
         bias_encoder_dropout_rate: float = 0.0,
+        preencoder: Optional[AbsPreEncoder] = None,
+        postencoder: Optional[AbsPostEncoder] = None,
     ):
         assert check_argument_types()
         assert 0.0 <= ctc_weight <= 1.0, ctc_weight

+ 1 - 1
funasr/models/e2e_asr_mfcca.py

@@ -53,7 +53,7 @@ class MFCCA(FunASRModel):
             encoder: AbsEncoder,
             decoder: AbsDecoder,
             ctc: CTC,
-            rnnt_decoder: None,
+            rnnt_decoder: None = None,
             ctc_weight: float = 0.5,
             ignore_id: int = -1,
             lsm_weight: float = 0.0,