mergekit_config.yml

merge_method: slerp
base_model: chanwit/flux-7b-base-stage-0
slices:
  - sources:
      - model: chanwit/flux-7b-base-stage-0
        layer_range: [0, 32]
      - model: meta-math/MetaMath-Mistral-7B
        layer_range: [0, 32]
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 # fallback for rest of tensors
dtype: bfloat16