ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix(studio): disable mlx gc for none (#5991)

Browse source
This commit is contained in:
Long Yixing 2026-06-04 15:38:45 +08:00 • committed by GitHub
commit 63dc27f76e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
studio/backend/core/training/worker.py
View file

@ -1362,7 +1362,7 @@ def _run_mlx_training(event_queue, stop_queue, config):
gc_setting = config.get("gradient_checkpointing", "mlx")
if isinstance(gc_setting, str):
use_grad_checkpoint = (
gc_setting if gc_setting.lower() not in ("false", "") else False
gc_setting if gc_setting.lower() not in ("false", "none", "") else False
)
else:
use_grad_checkpoint = gc_setting