| File was renamed from funasr/punctuation/vad_realtime_transformer.py |
| | |
| | | import torch.nn as nn |
| | | |
| | | from funasr.modules.embedding import SinusoidalPositionEncoder |
| | | from funasr.punctuation.sanm_encoder import SANMVadEncoder as Encoder |
| | | from funasr.punctuation.abs_model import AbsPunctuation |
| | | from funasr.models.encoder.sanm_encoder import SANMVadEncoder as Encoder |
| | | from funasr.train.abs_model import AbsPunctuation |
| | | |
| | | |
| | | class VadRealtimeTransformer(AbsPunctuation): |
| | | |
| | | """ |
| | | 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, |