tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

make always skip_move_to_device default as true (#3084)

Browse Source
This commit is contained in:
Wing Lian
2025-08-26 09:30:22 -04:00
committed by GitHub
parent c4c4b90638
commit e1131e9619
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/axolotl/utils/schemas/model.py
View File

@@ -71,10 +71,9 @@ class ModelInputConfig(BaseModel):
) )
experimental_skip_move_to_device: bool | None = Field( experimental_skip_move_to_device: bool | None = Field(
default=None, default=True,
json_schema_extra={ json_schema_extra={
"description": "Don't move the model to the device before sharding. " "description": "Don't move the model to the device before sharding. Set to `false` to revert to legacy behavior."
"This is an experimental feature that may be included in the future as the default."
}, },
) )