shixian.shi
2023-12-28 3e3eed19450b05953792a3dda2bdfe45b55849bc
funasr/models/bat/conformer_chunk_encoder.py
@@ -29,7 +29,7 @@
from funasr.models.transformer.utils.subsampling import TooShortUttError
from funasr.models.transformer.utils.subsampling import check_short_utt
from funasr.models.transformer.utils.subsampling import StreamingConvInput
from funasr.utils.register import register_class
from funasr.register import tables
@@ -312,7 +312,7 @@
        return x, cache
@register_class("encoder_classes", "ConformerChunkEncoder")
@tables.register("encoder_classes", "ConformerChunkEncoder")
class ConformerChunkEncoder(nn.Module):
    """Encoder module definition.
    Args: