From d3e57fdc50b4bca11c60cee59eb2e9a50a90884f Mon Sep 17 00:00:00 2001
From: Daniel <znsoft@163.com>
Date: 星期六, 22 四月 2023 05:14:02 +0800
Subject: [PATCH] Merge branch 'alibaba-damo-academy:main' into main
---
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