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:13:04 +08:00
parent 15ef54a3ce
commit a20157435e
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
funasr/train_utils/trainer_ds.py
View File

@ -347,7 +347,10 @@ class Trainer:
if self.use_lora:
lora_outdir = f"{self.output_dir}/lora-{ckpt_name}"
os.makedirs(lora_outdir, exist_ok=True)
model.llm.save_pretrained(lora_outdir)
if hasattr(model, "module"):
model.module.llm.save_pretrained(lora_outdir)
else:
model.llm.save_pretrained(lora_outdir)
filename = os.path.join(self.output_dir, ckpt_name)
torch.save(state, filename)