speech_asr 3 anos atrás
pai
commit
7ee716759b

+ 1 - 1
egs_modelscope/speaker_diarization/speech_diarization_eend-ola-en-us-callhome-8k/infer.py

@@ -2,7 +2,7 @@ from modelscope.pipelines import pipeline
 from modelscope.utils.constant import Tasks
 
 inference_diar_pipline = pipeline(
-    task=Tasks.auto_speech_recognition,
+    task=Tasks.speaker_diarization,
     model='damo/speech_diarization_eend-ola-en-us-callhome-8k',
     model_revision="v1.0.0",
 )

+ 0 - 3
funasr/bin/asr_inference_launch.py

@@ -234,9 +234,6 @@ def inference_launch(**kwargs):
     elif mode == "rnnt":
         from funasr.bin.asr_inference_rnnt import inference_modelscope
         return inference_modelscope(**kwargs)
-    elif mode == "eend-ola":
-        from funasr.bin.eend_ola_inference import inference_modelscope
-        return inference_modelscope(mode=mode, **kwargs)
     else:
         logging.info("Unknown decoding mode: {}".format(mode))
         return None