From 01f29c13253ee9b699daf37332e8ccdd016e6890 Mon Sep 17 00:00:00 2001 From: Kashif Rasul Date: Wed, 31 Jul 2024 21:23:10 +0200 Subject: [PATCH] remove revision (#186) --- scripts/run_cpt.py | 2 +- scripts/run_dpo.py | 2 +- scripts/run_sft.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) 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 ***")