Mellum

Mellum2 Instruct (SFT)

Research artifact: the SFT-only intermediate checkpoint of the Instruct training pipeline. Use it to study or build on the SFT stage in isolation, run post-training experiments (preference tuning, RLVR, etc.), or compare against the final RL-tuned variant. For production-quality instruction following use Instruct instead.

Mellum2 Instruct (SFT) Highlights

Mellum2 Instruct (SFT) is the supervised-fine-tuning-only intermediate checkpoint of the Mellum2 Instruct training pipeline, released by JetBrains as a research artifact.

The model uses a Mixture-of-Experts architecture with 64 experts and activates 8 experts per token. It uses a combination of sliding-window and full attention layers, with a context length of 131,072 tokens.

This checkpoint was produced from Mellum2-12B-A2.5B-Base with three epochs of SFT on a mixed corpus of coding, agentic, tool use, instruction following, general chat, identity, and safety data. It is the starting point of the RLVR stage that produces the final Mellum2-12B-A2.5B-Instruct.

Mellum2 Model Family

This repository contains one checkpoint from the Mellum2 family.

Checkpoint Description
Base Pretrain Base checkpoint before long-context extension
Base Final base model
Instruct SFT Supervised instruction-tuned checkpoint
Thinking SFT Supervised thinking checkpoint
Instruct RL-tuned instruction model
Thinking RL-tuned thinking model

Model Overview

Mellum2 Instruct (SFT) has the following features:

  • Number of Layers: 28
  • Hidden Size: 2304
  • Intermediate Size: 7168
  • MoE Intermediate Size: 896
  • Number of Experts: 64
  • Number of Activated Experts: 8
  • Number of Attention Heads (GQA): 32 for Q and 4 for KV
  • Context Length: 131,072
  • Sliding Window: 1,024
  • Vocabulary Size: 98,304
  • Precision: bfloat16

Serving with vLLM

# Without tool calling
vllm serve JetBrains/Mellum2-12B-A2.5B-Instruct-SFT --max-model-len 131072

# With tool calling
vllm serve JetBrains/Mellum2-12B-A2.5B-Instruct-SFT \
  --max-model-len 131072 \
  --enable-auto-tool-choice \
  --tool-call-parser hermes

Quickstart

Text-Only Input

from openai import OpenAI
# Configured by environment variables
client = OpenAI()

messages = [
    {"role": "user", "content": "Write a Python function to reverse a string."},
]

chat_response = client.chat.completions.create(
    model="JetBrains/Mellum2-12B-A2.5B-Instruct-SFT",
    messages=messages,
    max_tokens=81920,
    temperature=0.6,
    top_p=0.95,
    extra_body={
        "top_k": 20,
    },
)
print("Chat response:", chat_response)

Evaluation

Evaluation results are available in the model card. All values are self-reported by JetBrains.

For more details, see the Mellum2 Technical Report.

License

Released under the Apache 2.0 license.

Downloads last month
359
Safetensors
Model size
12B params
Tensor type
BF16
Β·
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support

Model tree for JetBrains/Mellum2-12B-A2.5B-Instruct-SFT

Quantizations
2 models

Spaces using JetBrains/Mellum2-12B-A2.5B-Instruct-SFT 3

Collection including JetBrains/Mellum2-12B-A2.5B-Instruct-SFT

Paper for JetBrains/Mellum2-12B-A2.5B-Instruct-SFT

Article mentioning JetBrains/Mellum2-12B-A2.5B-Instruct-SFT

Evaluation results