speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

bugfix python runtime

Browse Source
This commit is contained in:
游雁 2024-07-25 15:28:18 +08:00
parent a20157435e
commit 17e64505e8
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
funasr/models/llm_asr/model.py
View File

@ -937,7 +937,10 @@ class LLMASR4(nn.Module):
logging.info(f"use_lora: {llm_conf.get('use_lora', False)}")
if llm_conf.get("use_lora", False):
from omegaconf import OmegaConf
lora_conf = llm_conf.get("lora_conf", {})
lora_conf = OmegaConf.to_container(lora_conf, resolve=True)
from peft import get_peft_model, LoraConfig, TaskType, PeftConfig, PeftModel
lora_init_param_path = lora_conf.get("init_param_path", None)