System Info
- TensorRT-LLM: current
main
- Backend: PyTorch
- Model:
unsloth/Qwen3.8-27B-NVFP4
- Reproduced on: DGX Spark / SM121
Who can help?
No response
Information
Reproduction
Attempt to load the checkpoint through LLM(...) or trtllm-serve with the PyTorch backend and text-only multimodal execution (disable_mm_encoder=True).
The checkpoint is a dense Qwen3.8/Qwen3.5-family VLM with mixed compressed-tensors quantization: packed NVFP4 dense MLP weights, FP8 attention/GDN/lm_head weights, FP8 KV cache, and one MTP layer.
Expected behavior
The checkpoint should select Qwen3_5VLModel, preserve each config group's quantization, map packed compressed-tensors names to TRT-LLM parameters, construct a rowwise-FP8 lm_head, and generate coherent text with and without MTP.
Actual behavior
Current main has several independent blockers:
- Generic dense Qwen3.6/3.8 VLM configs are classified as Qwen-Image-Bench configs.
- The compressed-tensors parser rejects or loses multiple
config_groups.
- Packed NVFP4 names/scales are not normalized for the Qwen mapper.
- Dense MLP quant-config keys do not match the
_DenseMlpAdapter runtime module path.
- The explicit FP8
lm_head entry is dropped, which would cast the FP8 weight to bf16 and discard its scale.
Validation
With focused fixes for those concerns, an SM121 full-model smoke loaded 23.91 GiB of model weights, selected native trtllm::nvfp4_gemm, constructed lm_head as FP8_PER_CHANNEL_PER_TOKEN, and generated coherent 64-token responses for two prompts. The same stack also generated coherently with the checkpoint MTP layer enabled.
The changes will be proposed as small dependent PRs so each concern can be reviewed independently.
System Info
mainunsloth/Qwen3.8-27B-NVFP4Who can help?
No response
Information
transformerscheckpoint loaderReproduction
Attempt to load the checkpoint through
LLM(...)ortrtllm-servewith the PyTorch backend and text-only multimodal execution (disable_mm_encoder=True).The checkpoint is a dense Qwen3.8/Qwen3.5-family VLM with mixed compressed-tensors quantization: packed NVFP4 dense MLP weights, FP8 attention/GDN/lm_head weights, FP8 KV cache, and one MTP layer.
Expected behavior
The checkpoint should select
Qwen3_5VLModel, preserve each config group's quantization, map packed compressed-tensors names to TRT-LLM parameters, construct a rowwise-FP8lm_head, and generate coherent text with and without MTP.Actual behavior
Current
mainhas several independent blockers:config_groups._DenseMlpAdapterruntime module path.lm_headentry is dropped, which would cast the FP8 weight to bf16 and discard its scale.Validation
With focused fixes for those concerns, an SM121 full-model smoke loaded 23.91 GiB of model weights, selected native
trtllm::nvfp4_gemm, constructedlm_headasFP8_PER_CHANNEL_PER_TOKEN, and generated coherent 64-token responses for two prompts. The same stack also generated coherently with the checkpoint MTP layer enabled.The changes will be proposed as small dependent PRs so each concern can be reviewed independently.