[fix] fix paraformer search.py (wenet-e2e#2580)

* [fix] fix paraformer search.py * fix lint
ishine · Jul 22, 2024 · 08b031b · 08b031b
1 parent eae7656
commit 08b031b
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/wenet/paraformer/search.py b/wenet/paraformer/search.py
@@ -126,12 +126,13 @@ def gen_timestamps_from_peak(cif_peaks: List[int],
         else:
             split = fire_place[i] + MAX_TOKEN_DURATION
             times.append([fire_place[i] * frame_rate, split * frame_rate])
-    if num_frames - fire_place[-1] > START_END_THRESHOLD:
-        end = (num_frames + fire_place[-1]) * 0.5
-        times[-1][1] = end * frame_rate
-        times.append([end * frame_rate, num_frames * frame_rate])
-    else:
-        times[-1][1] = num_frames * frame_rate
+    if len(times) > 0:
+        if num_frames - fire_place[-1] > START_END_THRESHOLD:
+            end = (num_frames + fire_place[-1]) * 0.5
+            times[-1][1] = end * frame_rate
+            times.append([end * frame_rate, num_frames * frame_rate])
+        else:
+            times[-1][1] = num_frames * frame_rate
     return times