From 45451ca6d2908a9a7a1c7591d8bfebec6ae60cdf Mon Sep 17 00:00:00 2001 From: "Yisheng (Ethan) He" Date: Wed, 3 Sep 2025 19:00:00 +0800 Subject: [PATCH] Fix frame_length too short bug. --- engines/infer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/engines/infer.py b/engines/infer.py index 9b7b8f1..ffd6cfe 100644 --- a/engines/infer.py +++ b/engines/infer.py @@ -168,7 +168,7 @@ class Audio2ExpressionInfer(InferBase): frame_length = math.ceil(audio.shape[0] / ssr * 30) output_context = DEFAULT_CONTEXT.copy() - volume = librosa.feature.rms(y=audio, frame_length=int(1 / 30 * ssr), hop_length=int(1 / 30 * ssr))[0] + volume = librosa.feature.rms(y=audio, frame_length=min(int(1 / 30 * ssr), len(audio)), hop_length=int(1 / 30 * ssr))[0] if (volume.shape[0] > frame_length): volume = volume[:frame_length]