From a79ae159da35c0dbc9f32d224405eb462b5178a2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=9B=9E=E6=9D=91=E7=9A=84=E8=AF=B1=E6=83=91?=
 <1214585092@qq.com>
Date: Tue, 14 Oct 2025 12:54:18 +0800
Subject: [PATCH] fix model.py index out of range

---
 model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/model.py b/model.py
index a89defd..83b0cff 100644
--- a/model.py
+++ b/model.py
@@ -907,7 +907,7 @@ def inference(
                 ts_max = encoder_out_lens[i] - 4
                 for pred_token, pred_frame in pred:
                     _end = _start + len(list(pred_frame))
-                    if pred_token != 0:
+                    if pred_token != 0 and token_id < len(tokens):
                         ts_left = max((_start*60-30)/1000, 0)
                         ts_right = min((_end*60-30)/1000, (ts_max*60-30)/1000)
                         timestamp.append([tokens[token_id], ts_left, ts_right])