嘉渊
2023-04-25 b9d1425028e480aa2c8dbd3502207e443dcd2060
funasr/models/encoder/conformer_encoder.py
@@ -14,7 +14,6 @@
from typeguard import check_argument_types
from funasr.models.ctc import CTC
from funasr.models.encoder.abs_encoder import AbsEncoder
from funasr.modules.attention import (
    MultiHeadedAttention,  # noqa: H301
    RelPositionMultiHeadedAttention,  # noqa: H301
@@ -277,7 +276,7 @@
        return x, mask
class ConformerEncoder(AbsEncoder):
class ConformerEncoder(torch.nn.Module):
    """Conformer encoder module.
    Args: