Views
No views yet
| Attribute | Value |
|---|---|
| Base Model | mistralai/Mixtral-8x22B-Instruct-v0.1 |
| Total Parameters | 141B (Mixture-of-Experts) |
| Active Parameters | ~39B (top-2 of 8 experts per token) |
| Architecture | Sparse MoE Transformer |
| Quantization | NVFP4 (W4A4 with FP4 weights and dynamic FP4 activations) |
| Format | compressed-tensors (safetensors) |
| Precision | FP4 weights (group_size=16), FP8 scales, lm_head unquantized |
| Approx. Size | ~75 GB (down from ~282 GB in BF16) |
| Context Length | 65,536 tokens |
| License | Apache 2.0 |
1from vllm import LLM, SamplingParams
2from transformers import AutoTokenizer
3
4model_id = "enfuse/Mixtral-8x22B-Instruct-v0.1-NVFP4"
5tokenizer = AutoTokenizer.from_pretrained(model_id)
6
7llm = LLM(model=model_id, tensor_parallel_size=2)
8sampling_params = SamplingParams(temperature=0.7, top_p=0.9, max_tokens=512)
9
10messages = [
11 {"role": "user", "content": "Explain the benefits of mixture-of-experts architectures."},
12]
13
14prompt = tokenizer.apply_chat_template(messages, add_generation_prompt=True, tokenize=False)
15outputs = llm.generate([prompt], sampling_params)
16print(outputs[0].outputs[0].text)NVFP4lm_head, MoE gate layers (block_sparse_moe.gate)--apply_chat_template --fewshot_as_multiturn, tensor_parallel_size=2 on NVIDIA B200 GPUs.| Benchmark | Metric | n-shot | NVFP4 | BF16 Reference | Recovery |
|---|---|---|---|---|---|
| ARC-Challenge | acc_norm | 25 | 59.30 | 72.7 | 81.6% |
| GSM8K | exact_match | 5 | 70.13 | 82.0 | 85.5% |
| HellaSwag | acc_norm | 10 | 81.00 | 89.1 | 90.9% |
| MMLU | acc | 5 | 68.80 | 77.8 | 88.4% |
| TruthfulQA MC2 | acc | 0 | 62.52 | 68.1 | 91.8% |
| Winogrande | acc | 5 | 76.09 | 85.2 | 89.3% |