From e1131e9619f9c86cdd8f2fec1774e41354972238 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Tue, 26 Aug 2025 09:30:22 -0400 Subject: [PATCH] make always skip_move_to_device default as true (#3084) --- src/axolotl/utils/schemas/model.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/axolotl/utils/schemas/model.py b/src/axolotl/utils/schemas/model.py index 56b206b51..04312eedd 100644 --- a/src/axolotl/utils/schemas/model.py +++ b/src/axolotl/utils/schemas/model.py @@ -71,10 +71,9 @@ class ModelInputConfig(BaseModel): ) experimental_skip_move_to_device: bool | None = Field( - default=None, + default=True, json_schema_extra={ - "description": "Don't move the model to the device before sharding. " - "This is an experimental feature that may be included in the future as the default." + "description": "Don't move the model to the device before sharding. Set to `false` to revert to legacy behavior." }, )