From 85361322923e83cea8918ce006c69477aefa8280 Mon Sep 17 00:00:00 2001 From: Dominic789654 <47224289+Dominic789654@users.noreply.github.com> Date: Sun, 11 Feb 2024 01:33:32 +0800 Subject: [PATCH] fix merge lora bug --- examples/merge_lora.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/merge_lora.py b/examples/merge_lora.py index 7c4a722e3..b27f75339 100644 --- a/examples/merge_lora.py +++ b/examples/merge_lora.py @@ -39,7 +39,7 @@ def main(): model_args, merge_lora_args = parser.parse_args_into_dataclasses() model_args.use_lora = True - model = AutoModel.get_model(model_args) + model = AutoModel.get_model(model_args, tune_strategy='none') model.merge_lora_weights() model.save(merge_lora_args.output_model_path, save_full_model=True)