[fix] merge lora fix

OptimalScale · Oct 15, 2024 · 019a361 · 019a361
1 parent 94021d3
commit 019a361
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -24,6 +24,7 @@ data/
 
 # output models
 output_models/
+adapter_model/
 
 # Distribution / packaging
 .Python

diff --git a/examples/merge_lora.py b/examples/merge_lora.py
@@ -66,6 +66,7 @@ def main():
         device=merge_lora_args.device,
         ds_config=merge_lora_args.ds_config
     )
+    model.activate_model_for_inference()
     model.merge_lora_weights()
     model.save(merge_lora_args.output_model_path, save_full_model=True)