File size: 564 Bytes
1779442 |
1 2 3 4 5 6 7 8 9 10 11 12 |
sparsity_stage:
sparsity_modifiers:
SparseGPTModifier: {sparsity: 0.33, mask_structure: '0:0', sequential_update: false}
run_type: !!python/object/apply:llmcompressor.recipe.stage.StageRunType [oneshot]
finetuning_stage:
finetuning_modifiers:
ConstantPruningModifier:
targets: ['re:.*q_proj.weight', 're:.*k_proj.weight', 're:.*v_proj.weight', 're:.*o_proj.weight',
're:.*gate_proj.weight', 're:.*up_proj.weight', 're:.*down_proj.weight']
start: 0
run_type: !!python/object/apply:llmcompressor.recipe.stage.StageRunType [train]
|