嘉渊
2023-05-10 ad55e73198f51786c8f2f4a7dda5b34fd790de3f
funasr/models/target_delay_transformer.py
@@ -6,13 +6,10 @@
import torch.nn as nn
from funasr.modules.embedding import SinusoidalPositionEncoder
#from funasr.models.encoder.transformer_encoder import TransformerEncoder as Encoder
from funasr.punctuation.sanm_encoder import SANMEncoder as Encoder
#from funasr.modules.mask import subsequent_n_mask
from funasr.train.abs_model import AbsPunctuation
from funasr.models.encoder.sanm_encoder import SANMEncoder as Encoder
class TargetDelayTransformer(AbsPunctuation):
class TargetDelayTransformer(torch.nn.Module):
    def __init__(
        self,