| | |
|---|
| | | |
|---|
| | | from funasr.datasets.collate_fn import CommonCollateFn |
|---|
| | | from funasr.datasets.preprocessor import PuncTrainTokenizerCommonPreprocessor |
|---|
| | | from funasr.train.abs_model import AbsPunctuation |
|---|
| | | from funasr.train.abs_model import PunctuationModel |
|---|
| | | from funasr.models.target_delay_transformer import TargetDelayTransformer |
|---|
| | | from funasr.models.vad_realtime_transformer import VadRealtimeTransformer |
|---|
| | |
|---|
| | | punc_choices = ClassChoices( |
|---|
| | | "punctuation", |
|---|
| | | classes=dict(target_delay=TargetDelayTransformer, vad_realtime=VadRealtimeTransformer), |
|---|
| | | type_check=AbsPunctuation, |
|---|
| | | default="target_delay", |
|---|
| | | ) |
|---|
| | | |
|---|
