pre commit hooks
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user