Views
No views yet
vllm serve nm-testing/Qwen3-14B-FP8-block --tensor_parallel_size 11from openai import OpenAI
2
3# Modify OpenAI's API key and API base to use vLLM's API server.
4openai_api_key = "EMPTY"
5openai_api_base = "http://<your-server-host>:8000/v1"
6
7client = OpenAI(
8 api_key=openai_api_key,
9 base_url=openai_api_base,
10)
11
12model = "nm-testing/Qwen3-14B-FP8-block"
13
14messages = [
15 {"role": "user", "content": "Explain quantum mechanics clearly and concisely."},
16]
17
18
19outputs = client.chat.completions.create(
20 model=model,
21 messages=messages,
22)
23
24generated_text = outputs.choices[0].message.content
25print(generated_text)1from transformers import AutoProcessor, Qwen3ForCausalLM
2
3from llmcompressor import oneshot
4from llmcompressor.modeling import replace_modules_for_calibration
5from llmcompressor.modifiers.quantization import QuantizationModifier
6
7MODEL_ID = "nm-testing/Qwen3-14B-FP8-block"
8
9# Load model.
10model = Qwen3ForCausalLM.from_pretrained(MODEL_ID, dtype="auto")
11processor = AutoProcessor.from_pretrained(MODEL_ID)
12model = replace_modules_for_calibration(model)
13
14# Configure the quantization algorithm and scheme.
15# In this case, we:
16# * quantize the weights to fp8 with per-block quantization
17# * quantize the activations to fp8 with dynamic token activations
18recipe = QuantizationModifier(
19 targets="Linear",
20 scheme="FP8_BLOCK",
21 ignore=["lm_head"],
22)
23
24# Apply quantization.
25oneshot(model=model, recipe=recipe)
26
27# Save to disk in compressed-tensors format.
28SAVE_DIR = MODEL_ID.rstrip("/").split("/")[-1] + "-FP8-block"
29model.save_pretrained(SAVE_DIR)
30processor.save_pretrained(SAVE_DIR)lm_eval \
--model vllm \
--model_args pretrained="nm-testing/Qwen3-14B-FP8-block",dtype=auto,add_bos_token=True,max_model_len=16384,tensor_parallel_size=1,gpu_memory_utilization=0.9,enable_chunked_prefill=True,trust_remote_code=True \
--tasks openllm \
--write_out \
--batch_size auto \
--show_configlm_eval \
--model vllm \
--model_args pretrained="nm-testing/Qwen3-14B-FP8-block",dtype=auto,add_bos_token=False,max_model_len=16384,tensor_parallel_size=1,gpu_memory_utilization=0.7,disable_log_stats=True,enable_chunked_prefill=True,trust_remote_code=True \
--tasks leaderboard \
--apply_chat_template \
--fewshot_as_multiturn \
--write_out \
--batch_size auto \
--show_configevalplus.evaluate --model "nm-testing/Qwen3-14B-FP8-block" \
--dataset "humaneval" \
--backend vllm \
--tp 1 \
--greedy
evalplus.evaluate --model "nm-testing/Qwen3-14B-FP8-block" \
--dataset "mbpp" \
--backend vllm \
--tp 1 \
--greedy| Category | Metric | Qwen/Qwen3-14B | nm-testing/Qwen3-14B-FP8-block | Recovery (%) |
|---|---|---|---|---|
| OpenLLM V1 | ARC-Challenge (Acc-Norm, 25-shot) | 69.71 | 69.80 | 100.12 |
| GSM8K (Strict-Match, 5-shot) | 88.40 | 88.40 | 100.00 | |
| HellaSwag (Acc-Norm, 10-shot) | 79.63 | 79.52 | 99.86 | |
| MMLU (Acc, 5-shot) | 78.85 | 78.73 | 99.85 | |
| TruthfulQA (MC2, 0-shot) | 58.58 | 58.76 | 100.30 | |
| Winogrande (Acc, 5-shot) | 73.56 | 74.27 | 100.97 | |
| Average Score | 74.79 | 74.91 | 100.16 | |
| OpenLLM V2 | IFEval (Inst Level Strict Acc, 0-shot) | 48.56 | 48.80 | 100.49 |
| BBH (Acc-Norm, 3-shot) | 31.64 | 30.76 | 97.20 | |
| Math-Hard (Exact-Match, 4-shot) | 19.79 | 19.56 | 98.85 | |
| GPQA (Acc-Norm, 0-shot) | 25.00 | 24.83 | 99.33 | |
| MUSR (Acc-Norm, 0-shot) | 39.02 | 39.15 | 100.34 | |
| MMLU-Pro (Acc, 5-shot) | 24.75 | 23.42 | 94.63 | |
| Average Score | 31.46 | 31.09 | 98.82 | |
| Coding | HumanEval pass@1 | 87.80 | 88.40 | 100.68 |
| HumanEval+ pass@1 | 84.80 | 85.40 | 100.71 | |
| MBPP pass@1 | 87.00 | 86.50 | 99.43 | |
| MBPP+ pass@1 | 75.40 | 74.10 | 98.28 |