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 +-
1 files changed, 1 insertions(+), 1 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:
--
Gitblit v1.9.1