| | |
|---|
| | | from funasr.models.decoder.transformer_decoder import ParaformerDecoderSAN |
|---|
| | | from funasr.models.decoder.transformer_decoder import TransformerDecoder |
|---|
| | | from funasr.models.decoder.contextual_decoder import ContextualParaformerDecoder |
|---|
| | | from funasr.models.e2e_asr import ESPnetASRModel |
|---|
| | | from funasr.models.e2e_asr import ASRModel |
|---|
| | | from funasr.models.e2e_asr_paraformer import Paraformer, ParaformerBert, BiCifParaformer, ContextualParaformer |
|---|
| | | from funasr.models.e2e_tp import TimestampPredictor |
|---|
| | | from funasr.models.e2e_asr_mfcca import MFCCA |
|---|
| | |
|---|
| | | model_choices = ClassChoices( |
|---|
| | | "model", |
|---|
| | | classes=dict( |
|---|
| | | asr=ESPnetASRModel, |
|---|
| | | asr=ASRModel, |
|---|
| | | uniasr=UniASR, |
|---|
| | | paraformer=Paraformer, |
|---|
| | | paraformer_bert=ParaformerBert, |
|---|
| | |
|---|
| | | token_type=args.token_type, |
|---|
| | | token_list=args.token_list, |
|---|
| | | bpemodel=args.bpemodel, |
|---|
| | | non_linguistic_symbols=args.non_linguistic_symbols, |
|---|
| | | non_linguistic_symbols=args.non_linguistic_symbols if hasattr(args, "non_linguistic_symbols") else None, |
|---|
| | | text_cleaner=args.cleaner, |
|---|
| | | g2p_type=args.g2p, |
|---|
| | | split_with_space=args.split_with_space if hasattr(args, "split_with_space") else False, |
|---|
