slices: | |
- sources: | |
- model: unsloth/gemma-2-9b-it | |
layer_range: [0, 24] | |
- sources: # add middle layers with residuals scaled to zero | |
- model: unsloth/gemma-2-9b-it | |
layer_range: [8, 24] | |
parameters: | |
scale: | |
- filter: o_proj | |
value: 0.0 | |
- filter: down_proj | |
value: 0.0 | |
- value: 1.0 | |
- sources: # add middle layers with residuals scaled to zero | |
- model: unsloth/gemma-2-9b-it | |
layer_range: [8, 28] | |
parameters: | |
scale: | |
- filter: o_proj | |
value: 0.0 | |
- filter: down_proj | |
value: 0.0 | |
- value: 1.0 | |
- sources: | |
- model: unsloth/gemma-2-9b-it | |
layer_range: [24, 28] | |
merge_method: passthrough | |
dtype: bfloat16 |