update repo

嘉渊

2023-05-12 cb7c35d9a6d9883afd6eb1995840f1f4d7a182fa

 funasr/models/encoder/conformer_encoder.py

@@ -19,6 +19,7 @@
    RelPositionMultiHeadedAttention,  # noqa: H301
    LegacyRelPositionMultiHeadedAttention,  # noqa: H301
)
from funasr.models.encoder.abs_encoder import AbsEncoder
from funasr.modules.embedding import (
    PositionalEncoding,  # noqa: H301
    ScaledPositionalEncoding,  # noqa: H301
@@ -41,6 +42,7 @@
from funasr.modules.subsampling import TooShortUttError
from funasr.modules.subsampling import check_short_utt
from funasr.modules.subsampling import Conv2dSubsamplingPad

class ConvolutionModule(nn.Module):
    """ConvolutionModule in Conformer model.

@@ -276,7 +278,7 @@
        return x, mask


class ConformerEncoder(torch.nn.Module):
class ConformerEncoder(AbsEncoder):
    """Conformer encoder module.

    Args: