From cc91b8fb60c31fbcf4b8479eead42ad0a16d35bb Mon Sep 17 00:00:00 2001
From: lyblsgo <lyblsgo@163.com>
Date: 星期一, 08 五月 2023 14:18:41 +0800
Subject: [PATCH] modify copyright

---
 funasr/bin/vad_inference.py |    6 ++----
 1 files changed, 2 insertions(+), 4 deletions(-)

diff --git a/funasr/bin/vad_inference.py b/funasr/bin/vad_inference.py
index 1e19f5f..f9dc397 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):
@@ -352,7 +352,6 @@
                 item = {'key': keys[i], 'value': results[i]}
                 vad_results.append(item)
                 if writer is not None:
-                    results[i] = json.loads(results[i])
                     ibest_writer["text"][keys[i]] = "{}".format(results[i])
 
         return vad_results
@@ -466,7 +465,6 @@
                         item = {'key': keys[i], 'value': results[i]}
                         vad_results.append(item)
                         if writer is not None:
-                            results[i] = json.loads(results[i])
                             ibest_writer["text"][keys[i]] = "{}".format(results[i])
 
         return vad_results

--
Gitblit v1.9.1