feat: always push checkpoint to hub if set (axolotl-ai-cloud#1049) [s…

…kip ci]
jinwonkim93 · Jan 5, 2024 · cbdbf9e · cbdbf9e
1 parent bdfefaf
commit cbdbf9e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/axolotl/core/trainer_builder.py b/src/axolotl/core/trainer_builder.py
@@ -611,6 +611,7 @@ def build(self, total_num_steps):
             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