IssueRaiser

New MTP models not supported

Post from Google about new MTP models: https://blog.google/innovation-and-ai/technology/developers-tools/multi-token-prediction-gemma-4/ HF link to models: https://huggingface.co/collections/mlx-community/gemma-4-assistant-mtp Simple script to download and prompt the model: ``` from mlx_lm import load, generate model, tokenizer = load("mlx-community/gemma-4-31B-it-assistant-bf16") messages = [{"role": "user", "content": "Explain quantum entanglement simply."}] prompt = tokenizer.apply_chat_template(messages, add_generation_prompt=True, tokenize=False) response = generate(model, tokenizer, prompt=prompt, max_tokens=512, verbose=True) ``` Resulting error: ValueError: Model type gemma4_assistant not supported.

Posted by danielringwald on May 8, 2026, 8:14 AM

Tags

Related issues

Raise real-world problems, find people facing the same issue, and discuss solutions.2,000 issues0 comments

github (Other)

New MTP models not supported

danielringwaldMay 8, 2026, 8:14 AM

Post from Google about new MTP models: https://blog.google/innovation-and-ai/technology/developers-tools/multi-token-prediction-gemma-4/ HF link to models: https://huggingface.co/collections/mlx-community/gemma-4-assistant-mtp Simple script to download and prompt the model: ``` from mlx_lm import load, generate model, tokenizer = load("mlx-community/gemma-4-31B-it-assistant-bf16") messages = [{"role": "user", "content": "Explain quantum entanglement simply."}] prompt = tokenizer.apply_chat_template(messages, add_generation_prompt=True, tokenize=False) response = generate(model, tokenizer, prompt=prompt, max_tokens=512, verbose=True) ``` Resulting error: ValueError: Model type gemma4_assistant not supported.

Comments (0)

No comments yet. Start the discussion.

Comment on this issue