From 2d078a26feae1f8b230e51ffbb9e521c85607c0d Mon Sep 17 00:00:00 2001
From: shixian.shi <shixian.shi@alibaba-inc.com>
Date: 星期四, 25 一月 2024 10:23:19 +0800
Subject: [PATCH] update umap import

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

diff --git a/funasr/models/transformer/model.py b/funasr/models/transformer/model.py
index e2367a7..7e40060 100644
--- a/funasr/models/transformer/model.py
+++ b/funasr/models/transformer/model.py
@@ -348,7 +348,7 @@
         scorers["ngram"] = ngram
         
         weights = dict(
-            decoder=1.0 - kwargs.get("decoding_ctc_weight"),
+            decoder=1.0 - kwargs.get("decoding_ctc_weight", 0.0),
             ctc=kwargs.get("decoding_ctc_weight", 0.0),
             lm=kwargs.get("lm_weight", 0.0),
             ngram=kwargs.get("ngram_weight", 0.0),

--
Gitblit v1.9.1