Views
No views yet
SentenceTransformer.merge)
added to Sentence Transformers. It merges the top-70% task deltas with sign election (ties), relative to sentence-transformers/all-MiniLM-L6-v2 as the base.sentence-transformers/all-MiniLM-L6-v2sentence-transformers/multi-qa-MiniLM-L6-cos-v1sentence-transformers/all-MiniLM-L6-v21from sentence_transformers import SentenceTransformer
2
3merged = SentenceTransformer.merge(
4 models=["sentence-transformers/all-MiniLM-L6-v2",
5 "sentence-transformers/multi-qa-MiniLM-L6-cos-v1"],
6 weights=[0.5, 0.5],
7 densities=[0.7, 0.7],
8 method="ties",
9 base_model="sentence-transformers/all-MiniLM-L6-v2",
10 output_path="all-MiniLM-L6-v2-multi-qa-ties-merge",
11 dtype="float32",
12)1from sentence_transformers import SentenceTransformer
2
3model = SentenceTransformer("yjoonjang/all-MiniLM-L6-v2-multi-qa-ties-merge")
4emb = model.encode(["Model merging combines fine-tuned checkpoints.",
5 "It often beats each individual model."])
6print(model.similarity(emb, emb))