GLM-5.3-Flash NVFP4 with NVFP4 MTP

This checkpoint keeps the NVFP4 backbone from RedHatAI/GLM-5.3-Flash-NVFP4 and replaces its FP8 block MTP layer with a weight-only NVFP4 MTP layer produced from zai-org/GLM-5.3-Flash.

The MTP expert and shared-expert projections are packed as NVFP4. MTP attention, router, fusion, normalization, and shared-head tensors remain BF16. MTP activations are not statically quantized because Transformers does not construct the MTP module during calibration.

Recipe

The backbone recipe is included as recipe.yaml:

default_stage:
  default_modifiers:
    QuantizationModifier:
      config_groups:
        experts:
          targets: ["re:.*mlp\\.experts\\..*(gate|up|down)_proj$"]
          weights:
            num_bits: 4
            type: float
            symmetric: true
            group_size: 16
            strategy: tensor_group
            dynamic: false
            scale_dtype: torch.float8_e4m3fn
            observer: memoryless_minmax
          input_activations:
            num_bits: 4
            type: float
            symmetric: true
            group_size: 16
            strategy: tensor_group
            dynamic: local
            scale_dtype: torch.float8_e4m3fn
            observer: static_minmax
      targets: [Linear]
      ignore:
      - "re:.*visual.*"
      - lm_head
      - "re:.*mlp\\.gate$"
      - "re:.*self_attn\\.indexer\\..*"

With MTP quantization support from llm-compressor PR #3118, the corresponding oneshot call is:

oneshot(
    model=model,
    processor=processor,
    recipe="recipe.yaml",
    dataset=dataset,
    max_seq_length=4096,
    num_calibration_samples=512,
    moe_calibrate_all_experts=True,
    data_collator=data_collator,
    output_dir="/workspace/models/GLM-5.3-Flash-NVFP4-MTP",
    mtp_scheme="NVFP4",
)

mtp_scheme="NVFP4" applies model-free weight quantization to the unloaded MTP layer after the calibrated backbone is saved.

Validation

  • 867 MTP projections packed as NVFP4
  • 18 MTP weights retained in BF16 by the GLM architecture policy
  • 2,623 MTP tensors indexed in model_mtp.safetensors
  • No stale native FP8 inverse scales
  • All NVFP4 global scales finite and positive

vLLM

vllm serve soyrsoyr/GLM-5.3-Flash-NVFP4-MTP \
  --tensor-parallel-size 4 \
  --no-enable-flashinfer-autotune \
  --tool-call-parser glm47 \
  --enable-auto-tool-choice \
  --reasoning-parser glm45 \
  --gpu-memory-utilization 0.85 \
  --disable-custom-all-reduce \
  --speculative-config '{"method":"mtp","num_speculative_tokens":5}'
Downloads last month
-
Safetensors
Model size
321B params
Tensor type
F32
·
BF16
·
U8
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for soyrsoyr/GLM-5.3-Flash-NVFP4-MTP

Quantized
(2)
this model

Collection including soyrsoyr/GLM-5.3-Flash-NVFP4-MTP