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." }, )