From 264b85da264e40a2abc4bb9fb33cdd68f2721a2c Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 26 Dec 2024 17:07:25 -0800 Subject: [PATCH] Update save.py --- unsloth/save.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/unsloth/save.py b/unsloth/save.py index 8db3b6dc35..d3ba1928c4 100644 --- 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,