pre commit hooks

This commit is contained in:
Rahul Tuli
2025-04-02 22:35:36 +00:00
committed by Wing Lian
parent 56ba66b60f
commit 8beb2f27ad
4 changed files with 40 additions and 26 deletions

View File

@@ -1,4 +1,4 @@
base_model: "nm-testing/llama2.c-stories42M-gsm8k-sparse-only-uncompressed"
base_model: "nm-testing/llama2.c-stories42M-gsm8k-sparse-only-uncompressed"
# TODO: change to
# base_model: neuralmagic/Sparse-Llama-3.1-8B-2of4
@@ -68,11 +68,11 @@ llmcompressor:
ConstantPruningModifier:
targets: [
're:.*q_proj.weight',
're:.*k_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
start: 0