make always skip_move_to_device default as true (#3084)

2025-08-26 09:30:22 -04:00
parent c4c4b90638
commit e1131e9619
1 changed files with 2 additions and 3 deletions
--- 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."
        },
    )