Does not load into vllm- layer 5 fusion issue

#1
by mancub - opened

ValueError: Detected some but not all shards of language_model.model.layers.5.self_attn.qkv_proj are quantized. All shards of fused layers to have the same precision.

ValueError: Detected some but not all shards of language_model.model.layers.5.self_attn.qkv_proj are quantized. All shards of fused layers to have the same precision.

Yeah, I have no idea.

It's Gemma-4's fused tensors architecture @#()$!%.

I wasted a lot of time past week trying to make a mixed auto-round quant (INT8+INT4) of Gemma-4-31B, and while I do have a PoC I can't really tell if it's working correctly or not yet, as I haven't invested hours of quantizing full quality model. That layer 5 keeps biting me back in the ass non-stop because of that fusing.

Sign up or log in to comment