From 6f713226dd2828e81ef73acc219b97e51eb87f63 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Thu, 30 Jan 2025 11:48:48 -0500 Subject: [PATCH] make save_safetensors: true the default (#2292) * make save_safetensors: true the default * revert change to model output check --- src/axolotl/utils/config/models/input/v0_4_1/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/utils/config/models/input/v0_4_1/__init__.py b/src/axolotl/utils/config/models/input/v0_4_1/__init__.py index dc8897863..f05b259b7 100644 --- a/src/axolotl/utils/config/models/input/v0_4_1/__init__.py +++ b/src/axolotl/utils/config/models/input/v0_4_1/__init__.py @@ -528,7 +528,7 @@ class ModelOutputConfig(BaseModel): output_dir: str = Field(default="./model-out") hub_model_id: Optional[str] = None hub_strategy: Optional[str] = None - save_safetensors: Optional[bool] = None + save_safetensors: Optional[bool] = True class MLFlowConfig(BaseModel):