Merge pull request #165 from alibaba-damo-academy/dev_cmz

zhifu gao

2023-02-28 a016617c7ec98ab9c7475ff7d3b6150b98d5beeb

 funasr/punctuation/target_delay_transformer.py

@@ -8,7 +8,7 @@
from funasr.modules.embedding import PositionalEncoding
from funasr.modules.embedding import SinusoidalPositionEncoder
#from funasr.models.encoder.transformer_encoder import TransformerEncoder as Encoder
from funasr.models.encoder.sanm_encoder import SANMEncoder as Encoder
from funasr.punctuation.sanm_encoder import SANMEncoder as Encoder
#from funasr.modules.mask import subsequent_n_mask
from funasr.punctuation.abs_model import AbsPunctuation

@@ -73,6 +73,9 @@
        y = self.decoder(h)
        return y, None

    def with_vad(self):
        return False

    def score(self, y: torch.Tensor, state: Any, x: torch.Tensor) -> Tuple[torch.Tensor, Any]:
        """Score new token.