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

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-07-19 18:33:04 +00:00
commit a4f665f437
1 changed files with 1 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
unsloth/save.py
View file

@ -4645,9 +4645,7 @@ def _offload_model_for_quantize_subprocess(model):
# disk/meta is NOT: accelerate keeps those parameters off the model
# entirely, so removing the hooks and calling .to("cpu") would try to
# materialize the whole checkpoint into RAM (or move meta tensors).
if not all(
t.isdigit() or t.startswith(("cuda", "xpu")) or t == "cpu" for t in targets
):
if not all(t.isdigit() or t.startswith(("cuda", "xpu")) or t == "cpu" for t in targets):
return None
if not any(t.isdigit() or t.startswith(("cuda", "xpu")) for t in targets):
return None # nothing on an accelerator: no GPU memory to reclaim