speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

test

Browse Source
This commit is contained in:
语帆 2024-02-22 16:28:59 +08:00
parent cfc18a9047
commit c1c337ef9a
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
funasr/frontends/default.py
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.stft import Stft
from funasr.frontends.utils.frontend import Frontend from funasr.frontends.utils.frontend import Frontend
from funasr.models.transformer.utils.nets_utils import make_pad_mask from funasr.models.transformer.utils.nets_utils import make_pad_mask
from funasr.register import tables
@tables.register("frontend_classes", "DefaultFrontend") @tables.register("frontend_classes", "DefaultFrontend")
class DefaultFrontend(nn.Module): class DefaultFrontend(nn.Module):
@ -40,6 +42,7 @@ class DefaultFrontend(nn.Module):
frontend_conf: Optional[dict] = None, frontend_conf: Optional[dict] = None,
apply_stft: bool = True, apply_stft: bool = True,
use_channel: int = None, use_channel: int = None,
**kwargs,
): ):
super().__init__() super().__init__()
if isinstance(fs, str): if isinstance(fs, str):