shixian.shi
2024-01-10 2d0c8274a6dd0db9d5f7a00e71401bf7b4d65553
funasr/models/scama/sanm_encoder.py
@@ -30,7 +30,7 @@
from funasr.models.ctc.ctc import CTC
from funasr.utils.register import register_class, registry_tables
from funasr.register import tables
class EncoderLayerSANM(nn.Module):
    def __init__(
@@ -154,7 +154,7 @@
        return x, cache
@register_class("encoder_classes", "SANMEncoderChunkOpt")
@tables.register("encoder_classes", "SANMEncoderChunkOpt")
class SANMEncoderChunkOpt(nn.Module):
    """
    Author: Speech Lab of DAMO Academy, Alibaba Group