|
|
@@ -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,
|
|
|
),
|