feat: always push checkpoint to hub if set (#1049) [skip ci]
Browse files
src/axolotl/core/trainer_builder.py
CHANGED
@@ -611,6 +611,7 @@ class HFCausalTrainerBuilder(TrainerBuilderBase):
|
|
611 |
training_arguments_kwargs["hub_model_id"] = self.cfg.hub_model_id
|
612 |
training_arguments_kwargs["push_to_hub"] = True
|
613 |
training_arguments_kwargs["hub_private_repo"] = True
|
|
|
614 |
|
615 |
if self.cfg.hub_strategy:
|
616 |
training_arguments_kwargs["hub_strategy"] = self.cfg.hub_strategy
|
|
|
611 |
training_arguments_kwargs["hub_model_id"] = self.cfg.hub_model_id
|
612 |
training_arguments_kwargs["push_to_hub"] = True
|
613 |
training_arguments_kwargs["hub_private_repo"] = True
|
614 |
+
training_arguments_kwargs["hub_always_push"] = True
|
615 |
|
616 |
if self.cfg.hub_strategy:
|
617 |
training_arguments_kwargs["hub_strategy"] = self.cfg.hub_strategy
|