speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update predictor_bias for onnx model

Browse Source
This commit is contained in:
shixian.shi 2023-04-13 14:33:44 +08:00
parent c54bc90d36
commit bf5eb6f5ac
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
funasr/runtime/python/libtorch/funasr_torch/paraformer_bin.py
View File

@ -23,7 +23,6 @@ class Paraformer():
batch_size: int = 1,
device_id: Union[str, int] = "-1",
plot_timestamp_to: str = "",
pred_bias: int = 1,
quantize: bool = False,
intra_op_num_threads: int = 1,
):
@ -48,7 +47,10 @@ class Paraformer():
self.batch_size = batch_size
self.device_id = device_id
self.plot_timestamp_to = plot_timestamp_to
self.pred_bias = pred_bias
if "predictor_bias" in config['model_conf'].keys():
self.pred_bias = config['model_conf']['predictor_bias']
else:
self.pred_bias = 0
def __call__(self, wav_content: Union[str, np.ndarray, List[str]], **kwargs) -> List:
waveform_list = self.load_data(wav_content, self.frontend.opts.frame_opts.samp_freq)

6
funasr/runtime/python/onnxruntime/funasr_onnx/paraformer_bin.py
View File

@ -23,7 +23,6 @@ class Paraformer():
batch_size: int = 1,
device_id: Union[str, int] = "-1",
plot_timestamp_to: str = "",
pred_bias: int = 1,
quantize: bool = False,
intra_op_num_threads: int = 4,
):
@ -47,7 +46,10 @@ class Paraformer():
self.ort_infer = OrtInferSession(model_file, device_id, intra_op_num_threads=intra_op_num_threads)
self.batch_size = batch_size
self.plot_timestamp_to = plot_timestamp_to
self.pred_bias = pred_bias
if "predictor_bias" in config['model_conf'].keys():
self.pred_bias = config['model_conf']['predictor_bias']
else:
self.pred_bias = 0
def __call__(self, wav_content: Union[str, np.ndarray, List[str]], **kwargs) -> List:
waveform_list = self.load_data(wav_content, self.frontend.opts.frame_opts.samp_freq)