diff --git a/funasr/frontends/default.py b/funasr/frontends/default.py index 15cc35a27..70638e29c 100644 --- a/funasr/frontends/default.py +++ b/funasr/frontends/default.py @@ -16,6 +16,8 @@ from funasr.frontends.utils.log_mel import LogMel from funasr.frontends.utils.stft import Stft from funasr.frontends.utils.frontend import Frontend from funasr.models.transformer.utils.nets_utils import make_pad_mask +from funasr.register import tables + @tables.register("frontend_classes", "DefaultFrontend") class DefaultFrontend(nn.Module): @@ -40,6 +42,7 @@ class DefaultFrontend(nn.Module): frontend_conf: Optional[dict] = None, apply_stft: bool = True, use_channel: int = None, + **kwargs, ): super().__init__() if isinstance(fs, str):