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
|