From cc1a85962a6a697ae665f292f0abffad4abd7513 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=B8=B8=E9=9B=81?= <zhifu.gzf@alibaba-inc.com>
Date: Thu, 25 Jul 2024 12:37:31 +0800
Subject: [PATCH] bugfix python runtime

---
 funasr/models/llm_asr/model.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/funasr/models/llm_asr/model.py b/funasr/models/llm_asr/model.py
index 23e869761..882a5b436 100644
--- a/funasr/models/llm_asr/model.py
+++ b/funasr/models/llm_asr/model.py
@@ -958,7 +958,11 @@ class LLMASR4(nn.Module):
             src_state = torch.load(init_param_path, map_location="cpu")
             flag = audio_adaptor.load_state_dict(src_state, strict=False)
             logging.info(f"Loading audio_adaptor ckpt: {init_param_path}, status: {flag}")
-
+        freeze = audio_adaptor_conf.get("freeze", True)
+        if freeze:
+            for name, param in audio_adaptor.named_parameters():
+                param.requires_grad = False
+            audio_adaptor.eval()
         self.audio_adaptor = audio_adaptor
 
         self.error_calculator = None