add save_only_model arg

src/axolotl/core/trainer_builder.py

@@ -1058,6 +1058,9 @@ class HFCausalTrainerBuilder(TrainerBuilderBase):
         if self.cfg.save_safetensors is not None:
             training_arguments_kwargs["save_safetensors"] = self.cfg.save_safetensors
 
+        if self.cfg.save_only_model is not None:
+            training_arguments_kwargs["save_only_model"] = self.cfg.save_only_model
+
         if self.cfg.sample_packing_eff_est:
             training_arguments_kwargs[
                 "sample_packing_efficiency"

src/axolotl/utils/config/models/input/v0_4_1/__init__.py

@@ -355,6 +355,7 @@ class ModelOutputConfig(BaseModel):
     hub_model_id: Optional[str] = None
     hub_strategy: Optional[str] = None
     save_safetensors: Optional[bool] = None
+    save_only_model: Optional[bool] = None
 
 
 class MLFlowConfig(BaseModel):