Merge branch 'main' into funasr1.0

modelscope · Jan 18, 2024 · 9091e7a · 9091e7a
2 parents 8977176 + b28f3c9
commit 9091e7a
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/funasr/models/fsmn_vad_streaming/model.py b/funasr/models/fsmn_vad_streaming/model.py
@@ -501,7 +501,9 @@ def forward(self, feats: torch.Tensor, waveform: torch.tensor, cache: dict = {},
 		#     self.AllResetDetection()
 		return segments
 
-	def init_cache(self, cache: dict = {}):
+
+	def init_cache(self, cache: dict = {}, **kwargs):
+
 		cache["frontend"] = {}
 		cache["prev_samples"] = torch.empty(0)
 		cache["encoder"] = {}
@@ -528,9 +530,9 @@ def inference(self,
 	              cache: dict = {},
 	              **kwargs,
 	              ):
-		# cache = kwargs.get("cache", {})
+
 		if len(cache) == 0:
-			self.init_cache(cache)
+			self.init_cache(cache, **kwargs)
 
 		meta_data = {}
 		chunk_size = kwargs.get("chunk_size", 60000) # 50ms