From 172a3152b42af36443ec6a0a39969471c35b893d Mon Sep 17 00:00:00 2001
From: maliubiao <maliubiao@gmail.com>
Date: 星期日, 29 十二月 2024 22:33:22 +0800
Subject: [PATCH] 允许 model.generate 使用bytes io, 以便不写入文件，节省io时间 (#2343)

---
 examples/industrial_data_pretraining/fsmn_vad_streaming/demo.py |    4 +---
 1 files changed, 1 insertions(+), 3 deletions(-)

diff --git a/examples/industrial_data_pretraining/fsmn_vad_streaming/demo.py b/examples/industrial_data_pretraining/fsmn_vad_streaming/demo.py
index 0f30a37..21ce0cb 100644
--- a/examples/industrial_data_pretraining/fsmn_vad_streaming/demo.py
+++ b/examples/industrial_data_pretraining/fsmn_vad_streaming/demo.py
@@ -9,11 +9,9 @@
 
 model = AutoModel(model="iic/speech_fsmn_vad_zh-cn-16k-common-pytorch")
 
-mm = model.model
-for p in mm.parameters():
-    print(f"{p.numel()}")
 res = model.generate(input=wav_file)
 print(res)
+
 # [[beg1, end1], [beg2, end2], .., [begN, endN]]
 # beg/end: ms
 

--
Gitblit v1.9.1