From 47cee7fd7e4e7b9bdaf059b63a9e6e45974eab2f Mon Sep 17 00:00:00 2001 From: qingy1337 Date: Fri, 20 Dec 2024 02:22:27 -0800 Subject: [PATCH] Update save.py warning message (#1425) * Update README.md Llama 3.3 + Reddit * Update README.md Apple ML Cross Entropy * Update README.md Removing double citation * Update save.py warning message --------- Co-authored-by: Michael Han <107991372+shimmyshimmer@users.noreply.github.com> --- unsloth/save.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/save.py b/unsloth/save.py index cf78bf5897..027d88b2fc 100644 --- a/unsloth/save.py +++ b/unsloth/save.py @@ -577,7 +577,7 @@ def unsloth_save_model( # max_ram = max(max_ram - W.nbytes, 0) else: # Save to Disk - logger.warning_once("We will save to Disk and not RAM now.") + logger.warning_once("\nWe will save to Disk and not RAM now.") filename = os.path.join(temporary_location, f"{name}.pt") torch.save(W, filename, pickle_module = pickle, pickle_protocol = pickle.HIGHEST_PROTOCOL,) # weights_only = True weirdly fails?