allenai/tmax-2b GGUF

GGUF quantizations of allenai/tmax-2b, a terminal-agent model trained with DPPO on top of Qwen 3.5 2B.

Important: MTP head excluded

The original allenai/tmax-2b checkpoint is a Qwen3_5ForConditionalGeneration model whose config.json declares mtp_num_hidden_layers: 1, but the shipped model.safetensors does not contain the MTP head weights. Converting with the default MTP settings produces a GGUF that fails to load with:

error loading model: missing tensor 'blk.24.attn_norm.weight'

These GGUFs were converted with --no-mtp so the nextn_predict_layers metadata is omitted and the model loads correctly in llama.cpp / LM Studio / any GGUF-compatible runtime that supports the qwen35 architecture. MTP speculative decoding is therefore not available for this checkpoint.

Quantizations

Name Size BPW Notes
Q8_0 1.9 GB 8.5 Highest quality quantized
Q6_K 1.4 GB 6.6 Very high quality
Q5_K_M 1.3 GB 5.7 High quality, good balance
Q4_K_M 1.2 GB 5.4 Recommended default
Q4_K_S 1.1 GB 4.9 Smaller than Q4_K_M
Q3_K_M 1.0 GB 4.3 Lower quality, smaller
Q2_K 924 MB 2.6 Smallest, significant quality loss

For most users, Q4_K_M is the recommended balance of quality and size.

Conversion details

  • Source: allenai/tmax-2b (BF16 safetensors, 3.5 GB)
  • Tool: llama.cpp convert_hf_to_gguf.py (build 9820) with --no-mtp --outtype bf16
  • Quantization: llama-quantize from BF16 to each target type
  • Architecture: qwen35 (hybrid linear/full attention, mRoPE)

Runtime requirements

Requires a GGUF runtime with qwen35 architecture support:

  • llama.cpp build >= ~9300 (b9334+ confirmed working)
  • LM Studio 0.4.12+ with llama.cpp runtime >= 2.5.1
  • Any runtime advertising Qwen 3.5 support

Usage

# llama.cpp
llama-cli -m tmax-2b-Q4_K_M.gguf -p "Hello" -n 256 -ngl 99

# LM Studio (CLI)
lms import tmax-2b-Q4_K_M.gguf -y -c --user-repo local/tmax-2b
lms load tmax-2b
lms chat tmax-2b

License

Apache 2.0, inherited from the base model. See allenai/tmax-2b for the original model card and responsible-use guidelines.

Downloads last month
126
GGUF
Model size
2B params
Architecture
qwen35
Hardware compatibility
Log In to add your hardware

2-bit

3-bit

4-bit

5-bit

6-bit

8-bit

Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for hudsongouge/tmax-2b-GGUF

Finetuned
Qwen/Qwen3.5-2B
Finetuned
allenai/tmax-2b
Quantized
(1)
this model