| | |
|---|
| | | os.environ['CUDA_VISIBLE_DEVICES'] = str(gpu_id) |
|---|
| | | inference_pipline = pipeline( |
|---|
| | | task=Tasks.auto_speech_recognition, |
|---|
| | | model="damo/speech_UniASR_asr_2pass-zh-cn-8k-common-vocab3445-pytorch-offline", |
|---|
| | | model="damo/speech_UniASR_asr_2pass-zh-cn-8k-common-vocab3445-pytorch-online", |
|---|
| | | output_dir=output_dir_job, |
|---|
| | | batch_size=1 |
|---|
| | | ) |
|---|
| | | audio_in = os.path.join(split_dir, "wav.{}.scp".format(idx)) |
|---|
| | | inference_pipline(audio_in=audio_in) |
|---|
| | | inference_pipline(audio_in=audio_in, param_dict={"decoding_model": "normal"}) |
|---|
| | | |
|---|
| | | |
|---|
| | | def modelscope_infer(params): |
|---|
