From 340c55838b56bb98508337f8404e0a74f22a20c2 Mon Sep 17 00:00:00 2001
From: gaochangfeng <54253717+gaochangfeng@users.noreply.github.com>
Date: 星期一, 22 七月 2024 15:28:27 +0800
Subject: [PATCH] EMO_UNK禁用和Merge VAD修复 (#1940)

---
 funasr/models/paraformer/cif_predictor.py |    2 +-
 1 files changed, 1 insertions(+), 1 deletions(-)

diff --git a/funasr/models/paraformer/cif_predictor.py b/funasr/models/paraformer/cif_predictor.py
index 535131f..24145cd 100644
--- a/funasr/models/paraformer/cif_predictor.py
+++ b/funasr/models/paraformer/cif_predictor.py
@@ -449,7 +449,7 @@
         mask = mask.transpose(-1, -2).float()
         mask = mask.squeeze(-1)
         hidden, alphas, token_num = self.tail_process_fn(hidden, alphas, mask=mask)
-        acoustic_embeds, cif_peak = cif_v1_export(hidden, alphas, self.threshold)
+        acoustic_embeds, cif_peak = cif_export(hidden, alphas, self.threshold)
 
         return acoustic_embeds, token_num, alphas, cif_peak
 

--
Gitblit v1.9.1