423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-15 16:18:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix src/llamafactory/train/callbacks.py

Browse Source 
Former-commit-id: c79a21aeaa5462770790887a6826d335e1ded5a2
This commit is contained in:
-.- 2024-07-10 12:05:51 +08:00
parent 025c8fe302
commit 18057e14ef
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/llamafactory/train/callbacks.py
View File

@ -79,7 +79,7 @@ def fix_valuehead_checkpoint(
if name.startswith("v_head."): if name.startswith("v_head."):
v_head_state_dict[name] = param v_head_state_dict[name] = param
else: else:
decoder_state_dict[name.replace("pretrained_model.", "", count=1)] = param decoder_state_dict[name.replace("pretrained_model.", "",1)] = param
model.pretrained_model.save_pretrained( model.pretrained_model.save_pretrained(
output_dir, state_dict=decoder_state_dict or None, safe_serialization=safe_serialization output_dir, state_dict=decoder_state_dict or None, safe_serialization=safe_serialization