From 44f86092ea4bc99f867f359e41061331bd5826dd Mon Sep 17 00:00:00 2001 From: Vaibhav Srivastav Date: Wed, 10 Jan 2024 19:19:32 +0530 Subject: [PATCH] Fix Tokenizer save error. (#278) --- lora/fuse.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lora/fuse.py b/lora/fuse.py index c2c9624a..ba34fe2a 100644 --- a/lora/fuse.py +++ b/lora/fuse.py @@ -66,7 +66,7 @@ if __name__ == "__main__": model.update_modules(tree_unflatten(fused_linears)) weights = dict(tree_flatten(model.parameters())) - utils.save_model(args.save_path, weights, tokenizer._tokenizer, config) + utils.save_model(args.save_path, weights, tokenizer, config) if args.upload_name is not None: hf_path = args.hf_path