From ad0039596c252c581a5aa8869c9435b65bca3ad0 Mon Sep 17 00:00:00 2001
From: 北念 <lzr265946@alibaba-inc.com>
Date: 星期四, 09 二月 2023 19:11:16 +0800
Subject: [PATCH] add BiCifParaformer
---
funasr/models/e2e_asr_paraformer.py | 2 --
funasr/utils/timestamp_tools.py | 1 -
funasr/bin/asr_inference_paraformer_vad_punc.py | 1 +
3 files changed, 1 insertions(+), 3 deletions(-)
diff --git a/funasr/bin/asr_inference_paraformer_vad_punc.py b/funasr/bin/asr_inference_paraformer_vad_punc.py
index 629ee4f..7d18e02 100644
--- a/funasr/bin/asr_inference_paraformer_vad_punc.py
+++ b/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'
diff --git a/funasr/models/e2e_asr_paraformer.py b/funasr/models/e2e_asr_paraformer.py
index 34ee35e..f8141cd 100644
--- a/funasr/models/e2e_asr_paraformer.py
+++ b/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(
diff --git a/funasr/utils/timestamp_tools.py b/funasr/utils/timestamp_tools.py
index 12337d1..33d1255 100644
--- a/funasr/utils/timestamp_tools.py
+++ b/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>':
--
Gitblit v1.9.1