funasr1.0

游雁

2024-01-04 e82bd3d226e392190fc5152b3b063137e76b6005

 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