FunASR

FunASR copied to clipboard

Process Process-6: Traceback (most recent call last): File "/root/miniconda3/envs/sensevoice/lib/python3.10/multiprocessing/process.py", line 314, in _bootstrap self.run() File "/root/miniconda3/envs/sensevoice/lib/python3.10/multiprocessing/process.py", line 108, in run self._target(*self._args, **self._kwargs) File "/usr/local/data2/workspace/egs_vocal_extractor/data/speech_det.py", line 156, in process_audio_task res = model.generate( File "/root/miniconda3/envs/sensevoice/lib/python3.10/site-packages/funasr/auto/auto_model.py", line 306, in generate return self.inference_with_vad(input, input_len=input_len, **cfg) File "/root/miniconda3/envs/sensevoice/lib/python3.10/site-packages/funasr/auto/auto_model.py", line 464, in inference_with_vad results = self.inference( File "/root/miniconda3/envs/sensevoice/lib/python3.10/site-packages/funasr/auto/auto_model.py", line 345, in inference res = model.inference(**batch, **kwargs) File "/root/miniconda3/envs/sensevoice/lib/python3.10/site-packages/funasr/models/sense_voice/model.py", line 931, in inference align = ctc_forced_align( File "/root/miniconda3/envs/sensevoice/lib/python3.10/site-packages/funasr/models/sense_voice/utils/ctc_alignment.py", line 45, in ctc_forced_align best_score[:, padding_num + 0] = log_probs[:, 0, blank] IndexError: index 2 is out of bounds for dimension 1 with size 2

以下是我调用的参数 model = AutoModel( model=model_dir, trust_remote_code=False, vad_model="fsmn-vad", vad_kwargs={"max_single_segment_time": 30000}, device=f"cuda:{device_id}", disable_update=True, disable_pbar=True, disable_log=True, )

res = model.generate( input=str(flac_path), cache={}, language="auto", # "zh", "en", "yue", "ja", "ko", "nospeech" use_itn=True, batch_size_s=60, merge_vad=True, # output_timestamp=True, merge_length_s=15, )