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-18 03:16:18 +00:00
commit cd0aee97a0
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
unsloth/save.py
View file

@ -1153,9 +1153,10 @@ def unsloth_save_model(
state_dict[f"model.layers.{j}.{item}.bias"] = bias
_dev_budget = _device_vram_budget(W.device)
if _dev_budget is not None and (
torch.cuda.memory_allocated(W.device) + W.nbytes
) < _dev_budget:
if (
_dev_budget is not None
and (torch.cuda.memory_allocated(W.device) + W.nbytes) < _dev_budget
):
# Fits on W's own GPU -> keep it there.
state_dict[name] = W
elif W.device.type != "cuda":