const_tag: &scale_factor 0.7071067812  # 1/sqrt(2) scaling for stability

attenuate-env: &attenuated_env
  parameters:
    scale:
      - filter: q_proj
        value: *scale_factor
      - filter: k_proj
        value: *scale_factor
      - filter: v_proj
        value: *scale_factor
      - filter: o_proj
        value: *scale_factor
      - value: 1.0

slices:
  # Preserve input layers
  - sources:
      - model: SvalTek/ColdBrew-Aphid
        layer_range: [0, 8]

  # expand upper layers
  - sources:
      - model: SvalTek/ColdBrew-Aphid
        layer_range: [8, 16]
  - sources:
      - model: SvalTek/ColdBrew-Aphid
        layer_range: [8, 16]
        <<: *attenuated_env

  # expand upper layers
  - sources:
      - model: SvalTek/ColdBrew-Aphid
        layer_range: [16, 22]
  - sources:
      - model: SvalTek/ColdBrew-Aphid
        layer_range: [16, 22]
        <<: *attenuated_env

  # Preserve output layers
  - sources:
      - model: SvalTek/ColdBrew-Aphid
        layer_range: [22, 27]

merge_method: passthrough
dtype: bfloat16