Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference time
Paper โข 2203.05482 โข Published โข 10
How to use sraj/Merge_SOUP_at0.1mlm with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("fill-mask", model="sraj/Merge_SOUP_at0.1mlm") # pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("sraj/Merge_SOUP_at0.1mlm")
model = AutoModelForMaskedLM.from_pretrained("sraj/Merge_SOUP_at0.1mlm", device_map="auto")This is a merge of pre-trained language models created using mergekit.
This model was merged using the Linear merge method.
The following models were included in the merge:
The following YAML configuration was used to produce this model:
models:
- model: sraj/CMB_Acad_LRDat0.1mlm
parameters:
weight: 1.0
- model: sraj/CMB_FWEdu_V3_LRDat0.1mlm
parameters:
weight: 1.0
- model: sraj/CMB_SYN_LRDat0.1mlm
parameters:
weight: 1.0
merge_method: linear
parameters:
normalize: true
dtype: bfloat16