tiny-random-Llama-NVFP4

A tiny random model for testing, shrunk from nvidia/Llama-3.1-8B-Instruct-FP4: the same architecture, quantization config and checkpoint layout at test sizes. Its key patterns, dtypes and tensor ranks match the real checkpoint's (scripts/extract_layout.py).

modelopt NVFP4 on every decoder linear (packed U8, E4M3 weight_scale per 16, F32 weight_scale_2, calibrated F32 input_scale) plus the FP8 KV cache's k_scale / v_scale. Quantized with modelopt's NVFP4QTensor.

reference/ holds the same weights dequantized to bf16, under the unquantized model's keys: the reference to compare logits against, so a test measures what the load path and kernels add, not the quantization itself.

The weights are random; the outputs mean nothing. scripts/ rebuilds it from the real checkpoint's config.json.

Downloads last month
207
Safetensors
Model size
1.12M params
Tensor type
BF16
·
U8
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support