From b75d1e89bb2f513a79bb07e9100ba1cd2bbcf40c Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期日, 09 六月 2024 00:32:57 +0800
Subject: [PATCH] fix bug
---
examples/industrial_data_pretraining/sense_voice/demo_fsmn.py | 3 ++-
1 files changed, 2 insertions(+), 1 deletions(-)
diff --git a/examples/industrial_data_pretraining/sense_voice/demo_fsmn.py b/examples/industrial_data_pretraining/sense_voice/demo_fsmn.py
index b94cba7..ce4bdf8 100644
--- a/examples/industrial_data_pretraining/sense_voice/demo_fsmn.py
+++ b/examples/industrial_data_pretraining/sense_voice/demo_fsmn.py
@@ -21,7 +21,8 @@
"language": "auto",
"fp16": True,
"gain_event": True,
+ "beam_size": 5,
}
-res = model.generate(input=input_wav, batch_size_s=0, DecodingOptions=DecodingOptions)
+res = model.generate(input=input_wav, batch_size_s=0, DecodingOptions=DecodingOptions, beam_size=5)
print(res)
--
Gitblit v1.9.1