From cbdbf9e6e5cd6e78c65ec2c79936b38200cd5835 Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Sat, 6 Jan 2024 03:09:42 +0900 Subject: [PATCH] feat: always push checkpoint to hub if set (#1049) [skip ci] --- src/axolotl/core/trainer_builder.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/axolotl/core/trainer_builder.py b/src/axolotl/core/trainer_builder.py index 1ca36eb41..465cfa1af 100644 --- a/src/axolotl/core/trainer_builder.py +++ b/src/axolotl/core/trainer_builder.py @@ -611,6 +611,7 @@ class HFCausalTrainerBuilder(TrainerBuilderBase): training_arguments_kwargs["hub_model_id"] = self.cfg.hub_model_id training_arguments_kwargs["push_to_hub"] = True training_arguments_kwargs["hub_private_repo"] = True + training_arguments_kwargs["hub_always_push"] = True if self.cfg.hub_strategy: training_arguments_kwargs["hub_strategy"] = self.cfg.hub_strategy