From 16edcb3be2c328f3377aff6555e6435b28980a52 Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Sun, 21 Jan 2024 04:13:03 +1100 Subject: [PATCH] Update save.py --- unsloth/save.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/unsloth/save.py b/unsloth/save.py index c266fdfcee..36a61aa5aa 100644 --- a/unsloth/save.py +++ b/unsloth/save.py @@ -796,7 +796,7 @@ def unsloth_save_pretrained_gguf( if push_to_hub: print("Unsloth: Uploading GGUF to Huggingface Hub...") username = upload_to_huggingface( - self, model, new_save_directory, token, + self, new_save_directory, token, "GGUF converted", "gguf", file_location, ) print(f"Saved to https://huggingface.co/{username}/{new_save_directory.lstrip('/')}") @@ -875,7 +875,7 @@ def unsloth_push_to_hub_gguf( print("Unsloth: Uploading GGUF to Huggingface Hub...") username = upload_to_huggingface( - self, model, new_save_directory, token, + self, new_save_directory, token, "GGUF converted", "gguf", file_location, ) print(f"Saved to https://huggingface.co/{username}/{new_save_directory.lstrip('/')}")