File size: 1,733 Bytes
d8e584e |
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 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
const_tag: &MODEL1 jukofyork/Dawn-Miqu-70B
const_tag: &MODEL2 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: *MODEL1
layer_range: [0, 10]
- sources:
- model: *MODEL1
layer_range: [10, 20]
<<: *scale_filter_env
- sources:
- model: *MODEL2
layer_range: [10, 20]
<<: *scale_filter_env
- sources:
- model: *MODEL1
layer_range: [20, 30]
<<: *scale_filter_env
- sources:
- model: *MODEL2
layer_range: [20, 30]
<<: *scale_filter_env
- sources:
- model: *MODEL1
layer_range: [30, 40]
<<: *scale_filter_env
- sources:
- model: *MODEL2
layer_range: [30, 40]
<<: *scale_filter_env
- sources:
- model: *MODEL1
layer_range: [40, 50]
<<: *scale_filter_env
- sources:
- model: *MODEL2
layer_range: [40, 50]
<<: *scale_filter_env
- sources:
- model: *MODEL1
layer_range: [50, 60]
<<: *scale_filter_env
- sources:
- model: *MODEL2
layer_range: [50, 60]
<<: *scale_filter_env
- sources:
- model: *MODEL1
layer_range: [60, 70]
<<: *scale_filter_env
- sources:
- model: *MODEL2
layer_range: [60, 70]
<<: *scale_filter_env
- sources:
- model: *MODEL1
layer_range: [70, 80]
merge_method: passthrough
dtype: float16 |