This commit is contained in:
语帆 2024-02-22 16:28:59 +08:00
parent cfc18a9047
commit c1c337ef9a

View File

@ -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):