l3b-sparse / recipe.yaml
semran1's picture
Upload folder using huggingface_hub
1779442 verified
raw
history blame contribute delete
564 Bytes
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]