funasr/models/e_branchformer/encoder.py
@@ -42,7 +42,7 @@ TooShortUttError, check_short_utt, ) from funasr.utils.register import register_class from funasr.register import tables class EBranchformerEncoderLayer(torch.nn.Module): """E-Branchformer encoder layer module. @@ -174,7 +174,7 @@ return x, mask @register_class("encoder_classes", "EBranchformerEncoder") @tables.register("encoder_classes", "EBranchformerEncoder") class EBranchformerEncoder(nn.Module): """E-Branchformer encoder module."""