From c5992ca03ea6d6c7b78e5c1d481a612d0f91ac21 Mon Sep 17 00:00:00 2001
From: hnluo <haoneng.lhn@alibaba-inc.com>
Date: 星期四, 27 四月 2023 01:47:46 +0800
Subject: [PATCH] Update asr_inference_paraformer_streaming.py
---
funasr/bin/vad_inference.py | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/funasr/bin/vad_inference.py b/funasr/bin/vad_inference.py
index 1e19f5f..387b622 100644
--- a/funasr/bin/vad_inference.py
+++ b/funasr/bin/vad_inference.py
@@ -109,7 +109,7 @@
fbanks, fbanks_len = self.frontend.forward_fbank(speech, speech_lengths)
feats, feats_len = self.frontend.forward_lfr_cmvn(fbanks, fbanks_len)
fbanks = to_device(fbanks, device=self.device)
- # feats = to_device(feats, device=self.device)
+ feats = to_device(feats, device=self.device)
feats_len = feats_len.int()
else:
raise Exception("Need to extract feats first, please configure frontend configuration")
@@ -131,7 +131,7 @@
"in_cache": in_cache
}
# a. To device
- batch = to_device(batch, device=self.device)
+ #batch = to_device(batch, device=self.device)
segments_part, in_cache = self.vad_model(**batch)
if segments_part:
for batch_num in range(0, self.batch_size):
--
Gitblit v1.9.1