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

Update save.py

Browse source
This commit is contained in:
Daniel Han 2025-10-21 11:33:40 -07:00
commit 59f2b2bb4c
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
unsloth/save.py
View file

@ -2606,8 +2606,9 @@ def unsloth_save_pretrained_torchao(
torchao_save_directory = save_directory + "-torchao"
# TorchAO does not support safe_serialization right now for 0.13.0. 0.14.0 yes!
safe_serialization = Version(importlib_version("torchao")) >= Version("0.14.0")
# TorchAO does not support safe_serialization right now 0.14.0 seems broken!
safe_serialization = Version(importlib_version("torchao")) > Version("0.14.0")
safe_serialization = False
if push_to_hub:
if token is None and push_to_hub: token = get_token()
model.push_to_hub(torchao_save_directory, safe_serialization = safe_serialization, token = token)