From 3df109adfccedeb134dea4ba2ea9a2da89872048 Mon Sep 17 00:00:00 2001
From: Isuxiz Slidder <48672727+Isuxiz@users.noreply.github.com>
Date: 星期一, 31 三月 2025 17:51:52 +0800
Subject: [PATCH] Update model.py to fix "IndexError: index 1 is out of bounds for dimension 1 with size 0" (#2454)
---
funasr/utils/load_utils.py | 30 +++++++++++++++++++++++-------
1 files changed, 23 insertions(+), 7 deletions(-)
diff --git a/funasr/utils/load_utils.py b/funasr/utils/load_utils.py
index aabf5d9..c38f1d9 100644
--- a/funasr/utils/load_utils.py
+++ b/funasr/utils/load_utils.py
@@ -10,7 +10,6 @@
import time
import logging
from torch.nn.utils.rnn import pad_sequence
-from pydub import AudioSegment
try:
from funasr.download.file import download_from_url
@@ -19,6 +18,11 @@
import pdb
import subprocess
from subprocess import CalledProcessError, run
+
+try:
+ from pydub import AudioSegment
+except:
+ pass
def is_ffmpeg_installed():
@@ -82,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,]
@@ -101,7 +107,8 @@
data_or_path_or_list
).squeeze() # [n_samples,]
elif data_type == "text" and tokenizer is not None:
- data_or_path_or_list = tokenizer.encode(data_or_path_or_list)
+ with open(data_or_path_or_list, "r") as f:
+ data_or_path_or_list = tokenizer.encode(f.read().strip())
elif data_type == "image": # undo
pass
elif data_type == "video": # undo
@@ -138,7 +145,10 @@
def load_bytes(input):
- input = validate_frame_rate(input)
+ try:
+ input = validate_frame_rate(input)
+ except:
+ pass
middle_data = np.frombuffer(input, dtype=np.int16)
middle_data = np.asarray(middle_data)
if middle_data.dtype.kind not in "iu":
@@ -153,16 +163,22 @@
array = np.frombuffer((middle_data.astype(dtype) - offset) / abs_max, dtype=np.float32)
return array
+
def validate_frame_rate(
input,
fs: int = 16000,
):
-
+
# 灏嗘枃浠惰鍙栦负瀛楄妭娴�
byte_data = BytesIO(input)
# 浣跨敤 pydub 鍔犺浇闊抽
- audio = AudioSegment.from_file(byte_data)
+ try:
+ audio = AudioSegment.from_file(byte_data)
+ except:
+ raise RuntimeError(
+ "You are decoding the pcm data, please install pydub first. via `pip install pydub`."
+ )
# 纭繚閲囨牱鐜囦负 16000 Hz
if audio.frame_rate != fs:
@@ -175,7 +191,7 @@
# 鑾峰彇閲嶆柊閲囨牱鍚庣殑瀛楄妭娴佹暟鎹�
input = output.read()
-
+
return input
--
Gitblit v1.9.1