From 765e6371bbfba54e4b08b65e12a53d33b3173c9c Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期三, 12 六月 2024 16:48:15 +0800
Subject: [PATCH] decoding
---
funasr/models/llm_asr/model.py | 2 +-
funasr/train_utils/load_pretrained_model.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/funasr/models/llm_asr/model.py b/funasr/models/llm_asr/model.py
index e7dd11e..f5d6a82 100644
--- a/funasr/models/llm_asr/model.py
+++ b/funasr/models/llm_asr/model.py
@@ -407,7 +407,7 @@
audio_encoder = encoder_class(input_size=input_size, **audio_encoder_conf)
audio_encoder_output_size = audio_encoder.output_size()
freeze = audio_encoder_conf.get("freeze", True)
- freeze_layer_num = audio_encoder_conf.get("freeze_layer_num", -1)
+ freeze_layer_num = int(audio_encoder_conf.get("freeze_layer_num", -1))
if freeze_layer_num > 0:
freeze_layer_num = range(freeze_layer_num)
else:
diff --git a/funasr/train_utils/load_pretrained_model.py b/funasr/train_utils/load_pretrained_model.py
index cb2d87e..6bada5f 100644
--- a/funasr/train_utils/load_pretrained_model.py
+++ b/funasr/train_utils/load_pretrained_model.py
@@ -60,7 +60,7 @@
if excludes is not None:
for k_ex in excludes:
if k.startswith(k_ex):
- logging.info(f"key: {{k}} matching: {k_ex}, excluded")
+ logging.info(f"key: {k} matching: {k_ex}, excluded")
excludes_flag = True
break
if excludes_flag:
--
Gitblit v1.9.1