Editing Models with Task Arithmetic
Paper
• 2212.04089 • Published
• 7
This is a merge of pre-trained language models created using mergekit.
There seemed to be an instability with a prior merge component that resulted in looping, so I went for a Magnum-adjacent model instead this time. This one seems to hold up for stability and coherence under temp=1.0, minP=0.02, and the usual formatting for this class of model.
This model was merged using the Task Arithmetic merge method using grimjim/mistralai-Mistral-Nemo-Base-2407 as a base.
The following models were included in the merge:
The following YAML configuration was used to produce this model:
base_model: grimjim/mistralai-Mistral-Nemo-Base-2407
dtype: bfloat16
merge_method: task_arithmetic
parameters:
normalize: true
models:
- model: grimjim/mistralai-Mistral-Nemo-Base-2407
- layer_range: [0, 10]
model: grimjim/mistralai-Mistral-Nemo-Instruct-2407
parameters:
weight: 1.00
- layer_range: [10, 36]
model: grimjim/AbMagnolia-v1-12B
parameters:
weight: 0.75
- layer_range: [10, 36]
model: Delta-Vector/Rei-V3-KTO-12B
parameters:
weight: 0.25
- layer_range: [36, 40]
model: grimjim/mistralai-Mistral-Nemo-Instruct-2407
parameters:
weight: 1.00