From 1028a8a036cabd6091fc1a040bbddd565fd3e911 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期三, 10 一月 2024 17:42:53 +0800
Subject: [PATCH] funasr1.0 paraformer_streaming WavFrontendOnline

---
 funasr/bin/inference.py |    5 ++++-
 1 files changed, 4 insertions(+), 1 deletions(-)

diff --git a/funasr/bin/inference.py b/funasr/bin/inference.py
index dedaf7d..c4ff69b 100644
--- a/funasr/bin/inference.py
+++ b/funasr/bin/inference.py
@@ -391,7 +391,10 @@
 			frontend = frontend_class(**kwargs["frontend_conf"])
 
 		self.frontend = frontend
+		if "frontend" in kwargs:
+			del kwargs["frontend"]
 		self.kwargs = kwargs
+
 	
 	def __call__(self, input, input_len=None, kwargs=None, **cfg):
 		
@@ -423,7 +426,7 @@
 			time2 = time.perf_counter()
 			meta_data["load_data"] = f"{time2 - time1:0.3f}"
 			speech, speech_lengths = extract_fbank(audio_sample_list, data_type=kwargs.get("data_type", "sound"),
-			                                       frontend=self.frontend)
+			                                       frontend=self.frontend, **kwargs)
 			time3 = time.perf_counter()
 			meta_data["extract_feat"] = f"{time3 - time2:0.3f}"
 			meta_data["batch_data_time"] = speech_lengths.sum().item() * self.frontend.frame_shift * self.frontend.lfr_n / 1000

--
Gitblit v1.9.1