Editing Models with Task Arithmetic
Paper • 2212.04089 • Published • 8
This is a merge of pre-trained language models created using mergekit.
This model was merged using the task arithmetic merge method using TheSkullery/llama-3-cat-8b-instruct-v1 as a base.
The following models were included in the merge:
The following YAML configuration was used to produce this model:
slices:
- sources:
- model: TheSkullery/llama-3-cat-8b-instruct-v1
layer_range: [0, 32]
- model: victunes/TherapyLlama-8B-v1
layer_range: [0, 32]
parameters:
weight: 0.25
- model: herisan/llama-3-8b_mental_health_counseling_conversations
layer_range: [0, 32]
parameters:
weight: 0.25
merge_method: task_arithmetic
base_model: TheSkullery/llama-3-cat-8b-instruct-v1
dtype: bfloat16