Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 12:10:09 +01:00
Code Issues Packages Projects Releases Wiki Activity

refine asr inference

Browse Source
This commit is contained in:
wenmeng.zwm
2023-09-18 14:58:14 +08:00
parent 1c173e905a
commit c50b2ba6f4
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
modelscope/pipelines/audio/asr_inference_pipeline.py
View File

@@ -319,6 +319,7 @@ class AutomaticSpeechRecognitionPipeline(Pipeline):
token_num_relax = None
decoding_ind = None
decoding_mode = None
fake_streaming = False
if os.path.exists(outputs['am_model_config']):
config_file = open(outputs['am_model_config'], encoding='utf-8')
root = yaml.full_load(config_file)
@@ -335,9 +336,7 @@ class AutomaticSpeechRecognitionPipeline(Pipeline):
decoding_ind = root['decoding_ind']
if 'decoding_mode' in root:
decoding_mode = root['decoding_mode']
if 'fake_streaming' in root:
fake_streaming = root['fake_streaming']
cmd['beam_size'] = root['beam_size']
cmd['penalty'] = root['penalty']
cmd['maxlenratio'] = root['maxlenratio']
@@ -354,6 +353,7 @@ class AutomaticSpeechRecognitionPipeline(Pipeline):
cmd['token_num_relax'] = token_num_relax
cmd['decoding_ind'] = decoding_ind
cmd['decoding_mode'] = decoding_mode
cmd['fake_streaming'] = fake_streaming
if outputs.__contains__('mvn_file'):
cmd['cmvn_file'] = outputs['mvn_file']
model_config = self.model_cfg['model_config']