Base Model: deepseek-ai/DeepSeek-V2-Lite (16B parameters)
Quantization: FP8_BLOCK (128x128 block structure)
Purpose: Demonstrates FP8 block quantization with weight padding for models with dimensions not evenly divisible by block size

Key Feature: Block Quantization Padding

DeepSeek-V2-Lite has intermediate_size=10944, which is not divisible by the block size of 128. This model uses weight padding to handle this:

Original intermediate_size: 10944
Padded intermediate_size: 11008 (86 × 128)

The padding is applied during quantization and the config.json reflects the padded dimensions for vLLM compatibility.

Usage with vLLM

python

from vllm import LLM, SamplingParams

llm = LLM(
    model="Etelis/DeepSeek-V2-Lite-FP8-BLOCK-padded",
    trust_remote_code=True,
    tensor_parallel_size=1,
)

sampling_params = SamplingParams(max_tokens=100, temperature=0.7)
output = llm.generate(["Hello, world!"], sampling_params)
print(output[0].outputs[0].text)

Requirements: H100 or newer GPU (SM 8.9+) for FP8 block quantization support.

Quantization Recipe

python

from transformers import AutoModelForCausalLM, AutoTokenizer
from llmcompressor import oneshot
from llmcompressor.modifiers.quantization import QuantizationModifier

MODEL_ID = "deepseek-ai/DeepSeek-V2-Lite"

model = AutoModelForCausalLM.from_pretrained(
    MODEL_ID, 
    torch_dtype="auto", 
    trust_remote_code=True, 
    device_map="auto"
)
tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, trust_remote_code=True)

# FP8 block quantization - ignore layers with composite dimensions
recipe = QuantizationModifier(
    targets="Linear",
    scheme="FP8_BLOCK",
    ignore=["lm_head", "re:.*kv_a_proj_with_mqa.*"]
)

oneshot(model=model, recipe=recipe)

model.save_pretrained("DeepSeek-V2-Lite-FP8-BLOCK-padded")
tokenizer.save_pretrained("DeepSeek-V2-Lite-FP8-BLOCK-padded")

Created With

llm-compressor (with padding support)
compressed-tensors (PR #547)

Ignored Layers

lm_head: Not quantized (standard practice)
kv_a_proj_with_mqa: Has composite dimensions (512 + 64 = 576) that cannot be safely padded

License

This model inherits the DeepSeek Model License from the base model.