diff --git a/funasr/models/lcbnet/model.py b/funasr/models/lcbnet/model.py index 09e6dd137..e83f8d783 100644 --- a/funasr/models/lcbnet/model.py +++ b/funasr/models/lcbnet/model.py @@ -428,7 +428,10 @@ class LCBNet(nn.Module): meta_data["load_data"] = f"{time2 - time1:0.3f}" pdb.set_trace() audio_sample_list = sample_list[0] - ocr_sample_list = sample_list[1] + if len(sample_list) >1: + ocr_sample_list = sample_list[1] + else: + ocr_sample_list = [294, 0] speech, speech_lengths = extract_fbank(audio_sample_list, data_type=kwargs.get("data_type", "sound"), frontend=frontend) time3 = time.perf_counter()