funasr/models/target_delay_transformer.py
@@ -10,7 +10,11 @@ class TargetDelayTransformer(torch.nn.Module): """ Author: Speech Lab of DAMO Academy, Alibaba Group CT-Transformer: Controllable time-delay transformer for real-time punctuation prediction and disfluency detection https://arxiv.org/pdf/2003.01309.pdf """ def __init__( self, vocab_size: int,