funasr1.0

游雁

2024-01-13 bdfd27b9e96bd55c449953bb577e1d4deeaf11c9

 funasr/models/ct_transformer/encoder.py

@@ -31,7 +31,7 @@

from funasr.models.ctc.ctc import CTC

from funasr.utils.register import register_class
from funasr.register import tables

class EncoderLayerSANM(nn.Module):
    def __init__(
@@ -155,7 +155,7 @@
        return x, cache


@register_class("encoder_classes", "SANMVadEncoder")
@tables.register("encoder_classes", "SANMVadEncoder")
class SANMVadEncoder(nn.Module):
    """
    Author: Speech Lab of DAMO Academy, Alibaba Group