Missing parameters

#1
by jhmonroe - opened

Trying to run this model in LM Studio yields the following error message regarding missing parameters:

🥲 Failed to load the model

Failed to load model.

Error when loading model: ValueError: Missing 54 parameters: 
language_model.model.layers.24.self_attn.k_norm.weight,
language_model.model.layers.24.self_attn.k_proj.weight,
language_model.model.layers.24.self_attn.v_proj.weight,
language_model.model.layers.25.self_attn.k_norm.weight,
language_model.model.layers.25.self_attn.k_proj.weight,
language_model.model.layers.25.self_attn.v_proj.weight,
language_model.model.layers.26.self_attn.k_norm.weight,
language_model.model.layers.26.self_attn.k_proj.weight,
language_model.model.layers.26.self_attn.v_proj.weight,
language_model.model.layers.27.self_attn.k_norm.weight,
language_model.model.layers.27.self_attn.k_proj.weight,
language_model.model.layers.27.self_attn.v_proj.weight,
language_model.model.layers.28.self_attn.k_norm.weight,
language_model.model.layers.28.self_attn.k_proj.weight,
language_model.model.layers.28.self_attn.v_proj.weight,
language_model.model.layers.29.self_attn.k_norm.weight,
language_model.model.layers.29.self_attn.k_proj.weight,
language_model.model.layers.29.self_attn.v_proj.weight,
language_model.model.layers.30.self_attn.k_norm.weight,
language_model.model.layers.30.self_attn.k_proj.weight,
language_model.model.layers.30.self_attn.v_proj.weight,
language_model.model.layers.31.self_attn.k_norm.weight,
language_model.model.layers.31.self_attn.k_proj.weight,
language_model.model.layers.31.self_attn.v_proj.weight,
language_model.model.layers.32.self_attn.k_norm.weight,
language_model.model.layers.32.self_attn.k_proj.weight,
language_model.model.layers.32.self_attn.v_proj.weight,
language_model.model.layers.33.self_attn.k_norm.weight,
language_model.model.layers.33.self_attn.k_proj.weight,
language_model.model.layers.33.self_attn.v_proj.weight,
language_model.model.layers.34.self_attn.k_norm.weight,
language_model.model.layers.34.self_attn.k_proj.weight,
language_model.model.layers.34.self_attn.v_proj.weight,
language_model.model.layers.35.self_attn.k_norm.weight,
language_model.model.layers.35.self_attn.k_proj.weight,
language_model.model.layers.35.self_attn.v_proj.weight,
language_model.model.layers.36.self_attn.k_norm.weight,
language_model.model.layers.36.self_attn.k_proj.weight,
language_model.model.layers.36.self_attn.v_proj.weight,
language_model.model.layers.37.self_attn.k_norm.weight,
language_model.model.layers.37.self_attn.k_proj.weight,
language_model.model.layers.37.self_attn.v_proj.weight,
language_model.model.layers.38.self_attn.k_norm.weight,
language_model.model.layers.38.self_attn.k_proj.weight,
language_model.model.layers.38.self_attn.v_proj.weight,
language_model.model.layers.39.self_attn.k_norm.weight,
language_model.model.layers.39.self_attn.k_proj.weight,
language_model.model.layers.39.self_attn.v_proj.weight,
language_model.model.layers.40.self_attn.k_norm.weight,
language_model.model.layers.40.self_attn.k_proj.weight,
language_model.model.layers.40.self_attn.v_proj.weight,
language_model.model.layers.41.self_attn.k_norm.weight,
language_model.model.layers.41.self_attn.k_proj.weight,
language_model.model.layers.41.self_attn.v_proj.weight.
MLX Community org

The quant loads and runs fine via mlx-lm and OptIQ Lab, those missing weights (k_norm/k_proj/v_proj for layers 24-41) are intentional in Gemma-4's architecture: those layers reuse KV from earlier layers (interleaved local/global attention), so the projections don't exist in the upstream model either. LM Studio's loader doesn't yet handle the Gemma-4 shared-KV layout and expects every layer to have its own k/v weights, same error should appear on mlx-community/gemma-4-e4b-it-4bit and Google's bf16 release.

see - https://github.com/lmstudio-ai/lmstudio-bug-tracker/issues/1803 and https://github.com/lmstudio-ai/lmstudio-bug-tracker/issues/1741

For now run it via mlx-optiq instead:

  pip install -U mlx-optiq                                                                                                                                 
  optiq serve --model mlx-community/gemma-4-e4b-it-OptiQ-4bit \
              --drafter mlx-community/gemma-4-e4b-it-assistant-bf16                                                                                        
                                                                                                                                                           

Demo GIF attached.
gemma4_e4b_lab_demo

Thank you!
All makes sense. Every time some new compression method comes out, it takes LM Studio a few days or weeks to catch up. Same thing happened with the last Gemma advances.
Can also confirm the errors DO NOT exist on the Optiq 26B and 31B versions. Those run in LM Studio just fine

One other question: WHy are the optiq models stripped of their vision capabilities?

MLX Community org

Answered it here https://huggingface.co/mlx-community/Qwen3.6-27B-OptiQ-4bit/discussions/1 but the main reason is that optiq is using mlx-lm for the actual quantization and it doesn't support the vision layers. At the moment the quants are meant for text only infernece.

Sign up or log in to comment

Free AI Image Generator No sign-up. Instant results. Open Now