Update save.py

2024-12-26 17:07:25 -08:00 · 2024-12-26 17:07:25 -08:00 · 264b85da26
commit 264b85da26
parent da4741bbef
1 changed files with 2 additions and 1 deletions
--- a/unsloth/save.py
+++ b/unsloth/save.py
@ -2131,7 +2131,8 @@ def unsloth_generic_save(
    if token is None and push_to_hub: token = get_token()
    merge_and_overwrite_lora(
        get_model_name,
-        model,
+        model                = model,
+        tokenizer            = tokenizer,
        save_directory       = save_directory,
        push_to_hub          = push_to_hub,
        private              = private,