| | |
|---|
| | | from funasr.models.target_delay_transformer import TargetDelayTransformer |
|---|
| | | from funasr.models.vad_realtime_transformer import VadRealtimeTransformer |
|---|
| | | from funasr.torch_utils.initialize import initialize |
|---|
| | | from funasr.train.abs_model import AbsPunctuation |
|---|
| | | from funasr.train.abs_model import PunctuationModel |
|---|
| | | from funasr.train.class_choices import ClassChoices |
|---|
| | | |
|---|
| | | punc_choices = ClassChoices( |
|---|
| | | "punctuation", |
|---|
| | | classes=dict(target_delay=TargetDelayTransformer, vad_realtime=VadRealtimeTransformer), |
|---|
| | | type_check=AbsPunctuation, |
|---|
| | | classes=dict( |
|---|
| | | target_delay=TargetDelayTransformer, |
|---|
| | | vad_realtime=VadRealtimeTransformer |
|---|
| | | ), |
|---|
| | | default="target_delay", |
|---|
| | | ) |
|---|
| | | model_choices = ClassChoices( |
|---|
