Nanobit commited on
Commit
cbdbf9e
1 Parent(s): bdfefaf

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