tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix model card upload for PEFT models (#1043)

Browse Source
This commit is contained in:
Hamel Husain
2024-01-04 18:13:54 -08:00
committed by GitHub
parent f243c2186d
commit 31d23504a5
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/axolotl/train.py
View File

@@ -188,6 +188,9 @@ def train(
if not cfg.hub_model_id: if not cfg.hub_model_id:
trainer.create_model_card(model_name=cfg.output_dir.lstrip("./")) trainer.create_model_card(model_name=cfg.output_dir.lstrip("./"))
elif cfg.hub_model_id:
# defensively push to the hub to ensure the model card is updated
trainer.push_to_hub()
return model, tokenizer return model, tokenizer