From 59f2b2bb4ca0d21e055e5d95973c09ed1515a607 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 21 Oct 2025 11:33:40 -0700 Subject: [PATCH] Update save.py --- unsloth/save.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/unsloth/save.py b/unsloth/save.py index a62d63fc86..4ca59bbda6 100644 --- a/unsloth/save.py +++ b/unsloth/save.py @@ -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)