From 1fa6aef222b33e059d8b6dc2ac477b7f68d9c434 Mon Sep 17 00:00:00 2001
From: SpenserCai <spensercai@gmail.com>
Date: Sat, 5 Oct 2024 10:40:20 +0800
Subject: [PATCH] update

---
 nodes/sensevoice_nodes.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/nodes/sensevoice_nodes.py b/nodes/sensevoice_nodes.py
index 3605e12..0e0207a 100644
--- a/nodes/sensevoice_nodes.py
+++ b/nodes/sensevoice_nodes.py
@@ -38,6 +38,9 @@ class SenseVoiceNode:
         source_sr = audio["sample_rate"]
         speech = fAudioTool.audio_resample(speech, source_sr)
         speech = fAudioTool.postprocess(speech)
+        # 判断语音长度是否大于30s
+        if speech.shape[1] > 30 * 22050 and use_fast_mode:
+            raise ValueError("Audio length is too long, please set use_fast_mode to False.")
         _, model_dir = download_sensevoice_small()
         model_arg = {
                 "input":speech[0],