diff --git a/scripts/run_cpt.py b/scripts/run_cpt.py index 5553c154..06c9e9d9 100644 --- a/scripts/run_cpt.py +++ b/scripts/run_cpt.py @@ -200,7 +200,7 @@ def main(): if training_args.push_to_hub is True: logger.info("Pushing to hub...") - trainer.push_to_hub(revision=training_args.hub_model_revision, **kwargs) + trainer.push_to_hub(**kwargs) logger.info("*** Training complete ***") diff --git a/scripts/run_dpo.py b/scripts/run_dpo.py index 8c944af7..972d969a 100644 --- a/scripts/run_dpo.py +++ b/scripts/run_dpo.py @@ -252,7 +252,7 @@ def main(): if training_args.push_to_hub is True: logger.info("Pushing to hub...") - trainer.push_to_hub(revision=training_args.hub_model_revision, **kwargs) + trainer.push_to_hub(**kwargs) logger.info("*** Training complete! ***") diff --git a/scripts/run_sft.py b/scripts/run_sft.py index 62c8493e..60a2dfdb 100644 --- a/scripts/run_sft.py +++ b/scripts/run_sft.py @@ -224,7 +224,7 @@ def main(): if training_args.push_to_hub is True: logger.info("Pushing to hub...") - trainer.push_to_hub(revision=training_args.hub_model_revision, **kwargs) + trainer.push_to_hub(**kwargs) logger.info("*** Training complete ***")