From 4c7bb22d912325e4f15e55fddad0fb436b3fea66 Mon Sep 17 00:00:00 2001
From: zhaoxf4 <zhaoxf4@qq.com>
Date: Wed, 17 Sep 2025 17:36:34 +0800
Subject: [PATCH] Update model.py
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

🐞 fix: batch inference error with timestamp
---
 model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/model.py b/model.py
index a89defd..d3bfe04 100644
--- a/model.py
+++ b/model.py
@@ -896,7 +896,7 @@ def inference(
                 align = ctc_forced_align(
                     logits_speech.unsqueeze(0).float(),
                     torch.Tensor(token_int[4:]).unsqueeze(0).long().to(logits_speech.device),
-                    (encoder_out_lens-4).long(),
+                    (encoder_out_lens-4).long()[i],
                     torch.tensor(len(token_int)-4).unsqueeze(0).long().to(logits_speech.device),
                     ignore_id=self.ignore_id,
                 )