From 9c5f00794e7c474d1079ff78e5c69e8c53a30e6e Mon Sep 17 00:00:00 2001
From: dudulu <solacowa@gmail.com>
Date: 星期一, 05 二月 2024 10:10:52 +0800
Subject: [PATCH] 修复chunk_size可能传入str或list结构 (#1357)
---
funasr/auto/auto_model.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/funasr/auto/auto_model.py b/funasr/auto/auto_model.py
index 3986a11..d99fc56 100644
--- a/funasr/auto/auto_model.py
+++ b/funasr/auto/auto_model.py
@@ -228,7 +228,7 @@
data_batch = data_list[beg_idx:end_idx]
key_batch = key_list[beg_idx:end_idx]
batch = {"data_in": data_batch, "key": key_batch}
- if (end_idx - beg_idx) == 1 and isinstance(data_batch[0], torch.Tensor): # fbank
+ if (end_idx - beg_idx) == 1 and kwargs.get("data_type", None) == "fbank": # fbank
batch["data_in"] = data_batch[0]
batch["data_lengths"] = input_len
--
Gitblit v1.9.1