From 583f918e0ec752518a0263b09ce9b9e55f047fa4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=AF=AD=E5=B8=86?= Date: Fri, 1 Mar 2024 11:18:29 +0800 Subject: [PATCH] atsr --- funasr/models/lcbnet/model.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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()