游雁
2024-01-04 36702d28297839f2b41be9a60e822a4059dbe3f6
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