From a0f03bd2a87d97d47a1636bbe6f0855a43160331 Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期三, 15 五月 2024 19:48:50 +0800
Subject: [PATCH] Dev gzf deepspeed (#1732)
---
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