From 3c349ac0531b07239f37b81254f8568ab80e3f6a Mon Sep 17 00:00:00 2001
From: Han Zhang <45134013+holazzer@users.noreply.github.com>
Date: 星期二, 18 三月 2025 11:45:37 +0800
Subject: [PATCH] fix: use converted token_ids for alignment for sensevoice model with timestamp output (#2429)
---
funasr/utils/load_utils.py | 4 +++-
1 files changed, 3 insertions(+), 1 deletions(-)
diff --git a/funasr/utils/load_utils.py b/funasr/utils/load_utils.py
index 1d80fcf..072b578 100644
--- a/funasr/utils/load_utils.py
+++ b/funasr/utils/load_utils.py
@@ -86,8 +86,10 @@
): # download url to local file
data_or_path_or_list = download_from_url(data_or_path_or_list)
- if isinstance(data_or_path_or_list, str) and os.path.exists(data_or_path_or_list): # local file
+ if (isinstance(data_or_path_or_list, str) and os.path.exists(data_or_path_or_list)) or hasattr(data_or_path_or_list, 'read'): # local file or bytes io
if data_type is None or data_type == "sound":
+ if hasattr(data_or_path_or_list, "read") and hasattr(data_or_path_or_list, "seek"):
+ data_or_path_or_list.seek(0)
# if use_ffmpeg:
# data_or_path_or_list = _load_audio_ffmpeg(data_or_path_or_list, sr=fs)
# data_or_path_or_list = torch.from_numpy(data_or_path_or_list).squeeze() # [n_samples,]
--
Gitblit v1.9.1