carlyou commited on
Commit
d7849a6
·
verified ·
1 Parent(s): d3be582

Add DeepSeek-V2-Lite quantized to NVFP4 via llm-compressor

Browse files
Files changed (1) hide show
  1. README.md +76 -0
README.md ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: other
3
+ license_name: deepseek-license
4
+ license_link: https://github.com/deepseek-ai/DeepSeek-V2/blob/main/LICENSE-MODEL
5
+ base_model: deepseek-ai/DeepSeek-V2-Lite
6
+ library_name: transformers
7
+ tags:
8
+ - deepseek
9
+ - mla
10
+ - moe
11
+ - nvfp4
12
+ - fp4
13
+ - quantized
14
+ - vllm
15
+ pipeline_tag: text-generation
16
+ model-index:
17
+ - name: DeepSeek-V2-Lite-NVFP4
18
+ results: []
19
+ ---
20
+
21
+ # DeepSeek-V2-Lite-NVFP4
22
+
23
+ NVFP4 (W4A4) quantized version of [deepseek-ai/DeepSeek-V2-Lite](https://huggingface.co/deepseek-ai/DeepSeek-V2-Lite), quantized using [llm-compressor](https://github.com/vllm-project/llm-compressor).
24
+
25
+ ## Model Details
26
+
27
+ | | |
28
+ |---|---|
29
+ | **Base model** | deepseek-ai/DeepSeek-V2-Lite (15.7B params) |
30
+ | **Architecture** | DeepseekV2ForCausalLM (MLA attention + MoE) |
31
+ | **Quantization** | NVFP4 — 4-bit floating point weights and activations |
32
+ | **Format** | compressed-tensors (nvfp4-pack-quantized) |
33
+ | **Size** | ~8.9 GB (3.5x compression from BF16) |
34
+ | **Group size** | 16 |
35
+ | **Scale dtype** | float8_e4m3fn |
36
+
37
+ ## Quantization Details
38
+
39
+ - **Method**: Post-training quantization (PTQ) via `llm-compressor` `oneshot`
40
+ - **Scheme**: NVFP4 — weights and input activations quantized to 4-bit float
41
+ - **Calibration**: 20 samples from [HuggingFaceFW/fineweb-edu](https://huggingface.co/datasets/HuggingFaceFW/fineweb-edu) (sample-10BT)
42
+ - **Ignored layers**: `lm_head` (kept in original precision)
43
+ - **Scales**: per-tensor global scale (FP32) + per-group local scale (FP8, group size 16)
44
+
45
+ ## Usage with vLLM
46
+
47
+ Requires a GPU with NVFP4 tensor core support (NVIDIA Blackwell, SM100+).
48
+
49
+ ```bash
50
+ vllm serve carlyou/DeepSeek-V2-Lite-NVFP4 \
51
+ --trust-remote-code \
52
+ --max-model-len 2048
53
+ ```
54
+
55
+ ```python
56
+ from vllm import LLM, SamplingParams
57
+
58
+ llm = LLM(
59
+ model="carlyou/DeepSeek-V2-Lite-NVFP4",
60
+ trust_remote_code=True,
61
+ max_model_len=2048,
62
+ )
63
+
64
+ output = llm.generate("Hello, world!", SamplingParams(max_tokens=128))
65
+ print(output[0].outputs[0].text)
66
+ ```
67
+
68
+ ## Intended Use
69
+
70
+ This model is primarily intended for benchmarking and testing NVFP4 quantization support in vLLM, particularly MLA attention + quantization fusion patterns on Blackwell GPUs.
71
+
72
+ ## Limitations
73
+
74
+ - Requires Blackwell GPU (SM100+) for FP4 tensor core acceleration
75
+ - Quantization may degrade output quality compared to FP8 or BF16 versions
76
+ - Not evaluated on standard benchmarks — use for testing/benchmarking only