From c553a8db1712c2a5deeef5bbb68bd1fdf8d61ab7 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期四, 13 六月 2024 17:38:01 +0800
Subject: [PATCH] decoding
---
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