嘉渊 2 лет назад
Родитель
Сommit
76a4049738
1 измененных файлов с 2 добавлено и 0 удалено
  1. 2 0
      funasr/build_utils/build_asr_model.py

+ 2 - 0
funasr/build_utils/build_asr_model.py

@@ -39,6 +39,7 @@ from funasr.models.encoder.mfcca_encoder import MFCCAEncoder
 from funasr.models.encoder.resnet34_encoder import ResNet34Diar
 from funasr.models.encoder.rnn_encoder import RNNEncoder
 from funasr.models.encoder.sanm_encoder import SANMEncoder, SANMEncoderChunkOpt
+from funasr.models.encoder.branchformer_encoder import BranchformerEncoder
 from funasr.models.encoder.transformer_encoder import TransformerEncoder
 from funasr.models.frontend.default import DefaultFrontend
 from funasr.models.frontend.default import MultiChannelFrontend
@@ -113,6 +114,7 @@ encoder_choices = ClassChoices(
         sanm=SANMEncoder,
         sanm_chunk_opt=SANMEncoderChunkOpt,
         data2vec_encoder=Data2VecEncoder,
+        branchformer=BranchformerEncoder,
         mfcca_enc=MFCCAEncoder,
         chunk_conformer=ConformerChunkEncoder,
     ),