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
607
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