File size: 964 Bytes
f1ec629 |
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 |
const_tag: &MODEL jukofyork/dark-miqu-70b
const_tag: &QK_ATTENUATION_FACTOR 0.8408964153 # sqrt(sqrt(1/2))
const_tag: &MLP_DOWN_SCALE_FACTOR 0.7071067812 # sqrt(1/2)
scale-filter-env: &scale_filter_env
parameters:
scale:
- filter: q_proj
value: *QK_ATTENUATION_FACTOR
- filter: k_proj
value: *QK_ATTENUATION_FACTOR
- filter: down_proj
value: *MLP_DOWN_SCALE_FACTOR
- value: 1.0
slices:
- sources:
- model: *MODEL
layer_range: [0, 20]
- sources:
- model: *MODEL
layer_range: [20, 40]
<<: *scale_filter_env
- sources:
- model: *MODEL
layer_range: [20, 40]
<<: *scale_filter_env
- sources:
- model: *MODEL
layer_range: [40, 60]
<<: *scale_filter_env
- sources:
- model: *MODEL
layer_range: [40, 60]
<<: *scale_filter_env
- sources:
- model: *MODEL
layer_range: [60, 80]
merge_method: passthrough
dtype: float16 |