语帆
2024-02-22 491b2af1ecdf26f1513ac6a83f3490bf1b265449
test
2个文件已修改
5 ■■■■ 已修改文件
funasr/auto/auto_model.py 3 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/frontends/default.py 2 ●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/auto/auto_model.py
@@ -153,12 +153,9 @@
        
        # build tokenizer
        tokenizer = kwargs.get("tokenizer", None)
        pdb.set_trace()
        if tokenizer is not None:
            tokenizer_class = tables.tokenizer_classes.get(tokenizer)
            pdb.set_trace()
            tokenizer = tokenizer_class(**kwargs["tokenizer_conf"])
            pdb.set_trace()
            kwargs["tokenizer"] = tokenizer
            kwargs["token_list"] = tokenizer.token_list
            vocab_size = len(tokenizer.token_list)
funasr/frontends/default.py
@@ -17,7 +17,7 @@
from funasr.frontends.utils.frontend import Frontend
from funasr.models.transformer.utils.nets_utils import make_pad_mask
@tables.register("frontend_classes", "DefaultFrontend")
class DefaultFrontend(nn.Module):
    """Conventional frontend structure for ASR.
    Stft -> WPE -> MVDR-Beamformer -> Power-spec -> Mel-Fbank -> CMVN