add BiCifParaformer

北念

2023-02-09 ad0039596c252c581a5aa8869c9435b65bca3ad0

add BiCifParaformer

 funasr/bin/asr_inference_paraformer_vad_punc.py

@@ -41,6 +41,7 @@
from funasr.tasks.vad import VADTask
from funasr.utils.timestamp_tools import time_stamp_lfr6, time_stamp_lfr6_pl
from funasr.bin.punctuation_infer import Text2Punc
from funasr.models.e2e_asr_paraformer import BiCifParaformer

header_colors = '\033[95m'
end_colors = '\033[0m'

 funasr/models/e2e_asr_paraformer.py

@@ -929,8 +929,6 @@
        ds_alphas, ds_cif_peak, us_alphas, us_cif_peak = self.predictor.get_upsample_timestamp(encoder_out,
                                                                                               encoder_out_mask,
                                                                                               token_num)

        import pdb; pdb.set_trace()
        return ds_alphas, ds_cif_peak, us_alphas, us_cif_peak

    def forward(

 funasr/utils/timestamp_tools.py

@@ -128,7 +128,6 @@
    res_txt = ""
    for char, timestamp in zip(char_list, timestamp_list):
        res_txt += "{} {} {};".format(char, timestamp[0], timestamp[1])
    logging.warning(res_txt)  # for test
    res = []
    for char, timestamp in zip(char_list, timestamp_list):
        if char != '<sil>':