verachen/Fun-ASR
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

no need to load weight here, uild_model of FunASR's AutoModel will do it

Browse Source
This commit is contained in:
Rex
2025-12-16 16:52:11 +08:00
parent a552da13e7
commit ee159c76c0
1 changed files with 0 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
model.py
View File

@ -116,13 +116,6 @@ class FunASRNano(nn.Module):
llm_dim if llm_dim is not None else audio_adaptor_conf["llm_dim"] llm_dim if llm_dim is not None else audio_adaptor_conf["llm_dim"]
) )
audio_adaptor = adaptor_class(**audio_adaptor_conf) audio_adaptor = adaptor_class(**audio_adaptor_conf)
init_param_path = audio_adaptor_conf.get("init_param_path", None)
if init_param_path is not None:
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", False) freeze = audio_adaptor_conf.get("freeze", False)
if freeze: if freeze:
for name, param in audio_adaptor.named_parameters(): for name, param in audio_adaptor.named_parameters():