mergekit_config.yml

slices:
  - sources:
      - model: distilbert/distilgpt2
        layer_range: [0, 6]
      - model: openai-community/gpt2
        layer_range: [0, 6]
merge_method: slerp
base_model: distilbert/distilgpt2
parameters:
  t:
    - filter: self_attn
      value: [0, 0.5, 0.3, 0.7, 1]
    - filter: mlp
      value: [1, 0.5, 0.7, 0.3, 0]
    - value: 0.5
dtype: bfloat16