Merge pull request #698 from Dominic789654/fix_merge_lora

fix merge lora bug
OptimalScale · Feb 12, 2024 · 467f0c9 · 467f0c9
2 parents 0e9a9a6 + 8536132
commit 467f0c9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)