File size: 1,052 Bytes
16fe11e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
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
|