first commit

2026-10-23 15:00:59 +08:00
commit 4627899899
156 changed files with 398139 additions and 0 deletions
--- a/dev/configs/w4a8_int/deepseek-r1.yml
+++ b/dev/configs/w4a8_int/deepseek-r1.yml
@@ -0,0 +1,70 @@
+model:
+  name: DeepSeek-R1
+  model_type: deepseek_v3
+
+paths:
+  checkpoint_dir: /home/mashun/vllm_project/checkpoints/DeepSeek-R1-0528-FP8
+  output_dir: /home/mashun/vllm_project/checkpoints/DeepSeek-R1-0528-w4a8
+
+runtime:
+  num_workers: 16
+  mode: dequant+quant # dequant | dequant+quant | quant (only support bf16)
+
+retry_files: ~
+
+quant:
+  version: 2
+  format:
+    TN:
+      - "re:model.layers.\\d+.mlp.experts.*"
+      - "re:model.layers.\\d+.self_attn.*"
+    NN: ~
+  batch_size: 64
+  block_size: 128
+  group_size: -1
+  symmetric: True
+  class:
+    int4:
+      - "re:model.layers.\\d+.mlp.experts.\\d+.down_proj.*"
+      - "re:model.layers.\\d+.mlp.experts.\\d+.gate_proj.*"
+      - "re:model.layers.\\d+.mlp.experts.\\d+.up_proj.*"
+    int8: 
+      - "re:.*shared_experts.*"
+      - "re:model.layers.\\d+.self_attn.q_a_proj.*"
+      - "re:model.layers.\\d+.self_attn.q_b_proj.*"
+      - "re:model.layers.\\d+.self_attn.kv_a_proj_with_mqa.*"
+      - "re:model.layers.\\d+.self_attn.kv_b_proj.*"
+      - "re:model.layers.\\d+.self_attn.o_proj.*"
+  ignore: 
+    - "re:.*embed_tokens.*"
+    - "re:.*input_layernorm.*"
+    - "re:.*post_attention_layernorm.*"
+    - "re:.*kv_a_layernorm.*"
+    - "re:.*q_a_layernorm.*"
+    - "re:.*eh_proj.*"
+    - "re:.*enorm.*"
+    - "re:.*hnorm.*"
+    - "re:model.layers.\\d+.mlp.gate_proj.*"
+    - "re:model.layers.\\d+.mlp.up_proj.*"
+    - "re:model.layers.\\d+.mlp.down_proj.*"
+    - "re:model.layers.\\d+.mlp.gate\\..*"
+    - "re:model.norm.*"
+    - "re:model.layers.\\d+.shared_head.norm.*"
+    - "re:.*lm_head.*"
+    - "re:model.layers.\\d+.shared_head.head.*"
+   
+dequant:
+  ignore:
+    - "re:.*embed_tokens.*"
+    - "re:.*input_layernorm.*"
+    - "re:.*post_attention_layernorm.*"
+    - "re:.*kv_a_layernorm.*"
+    - "re:.*q_a_layernorm.*"
+    - "re:.*eh_proj.*"
+    - "re:.*enorm.*"
+    - "re:.*hnorm.*"
+    - "re:model.layers.\\d+.mlp.gate\\..*"
+    - "re:model.norm.*"
+    - "re:model.layers.\\d+.shared_head.norm.*"
+    - "re:.*lm_head.*"
+    - "re:model.layers.\\d+.shared_head.head.*"