Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-25 12:39:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix demo service dataloader bug

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11049894

    * fix dataloader bug

* fix dataloader bug
This commit is contained in:
jiangyu.xzy
2022-12-10 01:02:36 +08:00
parent c8dcdd93da
commit 7aab9ca313
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -150,6 +150,7 @@ class AutomaticSpeechRecognitionPipeline(Pipeline):
cmd['token_num_relax'] = token_num_relax
cmd['decoding_ind'] = decoding_ind
cmd['decoding_mode'] = decoding_mode
cmd['num_workers'] = 0
elif self.framework == Frameworks.tf:
cmd['fs']['model_fs'] = inputs['model_config']['fs']
@@ -259,6 +260,7 @@ class AutomaticSpeechRecognitionPipeline(Pipeline):
minlenratio=cmd['minlenratio'],
beam_size=cmd['beam_size'],
ngpu=cmd['ngpu'],
num_workers=cmd['num_workers'],
ctc_weight=cmd['ctc_weight'],
lm_weight=cmd['lm_weight'],
penalty=cmd['penalty'],