frozenleaves/trl
1
0
Fork 0
mirror of https://github.com/huggingface/trl.git synced 2025-10-27 23:44:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

Standardize pushing to Hub in examples (#2126)

Browse Source
This commit is contained in:
Quentin Gallouédec
2024-09-26 10:00:51 +02:00
committed by GitHub
parent fb1b48fdbe
commit 32d9d34eb1
22 changed files with 88 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/scripts/rloo/rloo.py
View File

@ -129,7 +129,10 @@ if __name__ == "__main__":
eval_dataset=eval_dataset,
)
trainer.train()
# Save and push to hub
trainer.save_model(training_args.output_dir)
if training_args.push_to_hub:
trainer.push_to_hub()
trainer.generate_completions()