bi/embedding_model
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

merged model

Browse Source
This commit is contained in:
a.hediehloo 2025-11-23 08:40:24 +00:00
parent 4fb3504b33
commit 6aa4f8c0ea
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train/qwen/merge_model.py
View File

@ -15,8 +15,8 @@ def main():
file_path = os.path.dirname(__file__) file_path = os.path.dirname(__file__)
base_model_path = file_path + "/../../data/models/Qwen3-Embedding-0.6B/model" base_model_path = file_path + "/../../data/models/Qwen3-Embedding-0.6B/model"
peft_model_path = file_path + "/output/v0-20251118-115015/checkpoint-3434" peft_model_path = file_path + "/output/v1-20251122-184545/checkpoint-3434"
save_path = file_path + "/output/v0-20251118-115015/merged_checkpoint-3434" save_path = file_path + "/output/v1-20251122-184545/merged_checkpoint-3434"
merge(base_model_path, peft_model_path, save_path) merge(base_model_path, peft_model_path, save_path)