From 7c11a0d6a84b0f903a947f6a0e5a2730fb528c87 Mon Sep 17 00:00:00 2001
From: 雾聪 <wucong.lyb@alibaba-inc.com>
Date: 星期一, 27 五月 2024 19:15:58 +0800
Subject: [PATCH] adapt to new model struct
---
examples/industrial_data_pretraining/sense_voice/demo_fsmn.py | 1 +
1 files changed, 1 insertions(+), 0 deletions(-)
diff --git a/examples/industrial_data_pretraining/sense_voice/demo_fsmn.py b/examples/industrial_data_pretraining/sense_voice/demo_fsmn.py
index e063e1f..ce4bdf8 100644
--- a/examples/industrial_data_pretraining/sense_voice/demo_fsmn.py
+++ b/examples/industrial_data_pretraining/sense_voice/demo_fsmn.py
@@ -21,6 +21,7 @@
"language": "auto",
"fp16": True,
"gain_event": True,
+ "beam_size": 5,
}
res = model.generate(input=input_wav, batch_size_s=0, DecodingOptions=DecodingOptions, beam_size=5)
--
Gitblit v1.9.1