funasr/models/sanm/encoder.py @@ -484,7 +484,7 @@ return x, mask @tables.register("encoder_classes", "SANMEncoderChunkOptExport") @tables.register("encoder_classes", "SANMEncoderExport") class SANMEncoderExport(nn.Module): def __init__(