ivxxdegen commited on
Commit
c38e5cb
·
verified ·
1 Parent(s): 905bbfc

Create merge.py

Browse files
Files changed (1) hide show
  1. merge.py +16 -0
merge.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import AutoModelForCausalLM, PeftModel
2
+ from huggingface_hub import HfApi
3
+
4
+ # Load base model and adapter model
5
+ base_model = AutoModelForCausalLM.from_pretrained("microsoft/phi-4")
6
+ adapter_model = PeftModel.from_pretrained(base_model, "ivxxdegen/mibera-v1")
7
+
8
+ # Merge the LoRA adapter into the base model
9
+ merged_model = adapter_model.merge_and_unload()
10
+ merged_model.save_pretrained("merged_model")
11
+
12
+ # Upload the merged model to Hugging Face
13
+ api = HfApi()
14
+ api.upload_folder(folder_path="merged_model", repo_id="ivxxdegen/mibera-v1")
15
+
16
+ print("✅ Merge and upload completed successfully.")