setup

游雁

2023-05-24 d86b4531e0c64494f6a150e8220c7c6e7907d3dc

 funasr/build_utils/build_punc_model.py

@@ -3,14 +3,15 @@
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(