| | |
|---|
| | | |
|---|
| | | from funasr.utils.compute_wer import compute_wer |
|---|
| | | |
|---|
| | | import pdb; |
|---|
| | | def modelscope_infer_core(output_dir, split_dir, njob, idx): |
|---|
| | | output_dir_job = os.path.join(output_dir, "output.{}".format(idx)) |
|---|
| | | gpu_id = (int(idx) - 1) // njob |
|---|
| | |
|---|
| | | os.environ['CUDA_VISIBLE_DEVICES'] = str(gpu_id) |
|---|
| | | inference_pipline = pipeline( |
|---|
| | | task=Tasks.auto_speech_recognition, |
|---|
| | | model='yufan6/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950', |
|---|
| | | model_revision='v2.0.0', |
|---|
| | | model='NPU-ASLP/speech_mfcca_asr-zh-cn-16k-alimeeting-vocab4950', |
|---|
| | | model_revision='v3.0.0', |
|---|
| | | output_dir=output_dir_job, |
|---|
| | | batch_size=1, |
|---|
| | | ) |
|---|
