Mikhail Podvitskii 4fb16eccce model: add Mellum architecture (#23966) ...

* model: support for Mellum architecture

* model: improve mellum.py formatting

* model: improve mellum.py formatting once again

* deps: downgrade transformers to 4.57.6 (to fix CI)

* deps: remove huggingface_hub dependency

* deps: remove huggingface_hub from test requirements

---------

Co-authored-by: Sigbjørn Skjæret <sigbjorn.skjaeret@scala.com>

2026-06-02 22:11:12 +03:00

..

afmoe.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

apertus.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

arcee.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

arctic.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

arwkv7.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

baichuan.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

bailingmoe2.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

bailingmoe.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

bert.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

bitnet.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

bloom.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

chameleon.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

chatglm.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

codeshell.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

cogvlm.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

cohere2.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

command-r.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

dbrx.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

deci.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

deepseek2.cpp

model : fix model type check for granite/llama3 and deepseek2/glm4.7 lite (#22870)

2026-05-10 08:44:29 +02:00

deepseek2ocr.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

deepseek32.cpp

model : support for DeepseekV32ForCausalLM with generic DeepSeek Sparse Attention (DSA) implementation (#23346)

2026-05-29 10:15:17 +02:00

deepseek.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

delta-net-base.cpp

metal : optimize pad + cpy (#23354)

2026-05-20 09:42:00 +03:00

dots1.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

dream.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

ernie4-5-moe.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

ernie4-5.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

eurobert.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

exaone4.cpp

model: Add EXAONE 4.5 implementations (#21733)

2026-06-01 11:48:53 +02:00

exaone-moe.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

exaone.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

falcon-h1.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

falcon.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

gemma2.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

gemma3.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

gemma3n.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

gemma4.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

gemma-embedding.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

gemma.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

glm4-moe.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

glm4.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

glm-dsa.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

gpt2.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

gptneox.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

granite-hybrid.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

granite-moe.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

granite.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

grok.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

grovemoe.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

hunyuan-dense.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

hunyuan-moe.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

hunyuan-vl.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

internlm2.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

jais2.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

jais.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

jamba.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

jina-bert-v2.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

jina-bert-v3.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

kimi-linear.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

lfm2.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

lfm2moe.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

llada-moe.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

llada.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

llama4.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

llama-embed.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

llama.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

maincoder.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

mamba2.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

mamba-base.cpp

model : wire up Nemotron-H tensors for NVFP4 support (#20561)

2026-03-16 09:19:16 +01:00

mamba.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

mellum.cpp

model: add Mellum architecture (#23966)

2026-06-02 22:11:12 +03:00

mimo2.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

minicpm3.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

minicpm.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

minimax-m2.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

mistral3.cpp

models : Attach Mistral3 NVFP4 weight scales (#23629)

2026-05-26 07:59:59 +03:00

mistral4.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

models.h

model: add Mellum architecture (#23966)

2026-06-02 22:11:12 +03:00

modern-bert.cpp

model : support granite multilingual embeddings R2 (ibm-granite/granite-embedding-{97,311}m-multilingual-r2) (#22716)

2026-06-02 17:55:11 +02:00

mpt.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

nemotron-h-moe.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

nemotron-h.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

nemotron.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

neo-bert.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

nomic-bert-moe.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

nomic-bert.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

olmo2.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

olmo.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

olmoe.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

openai-moe.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

openelm.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

orion.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

paddleocr.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

pangu-embed.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

phi2.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

phi3.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

phimoe.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

plamo2.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

plamo3.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

plamo.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

plm.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

qwen2.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

qwen2moe.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

qwen2vl.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

qwen3.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

qwen3moe.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

qwen3next.cpp

llama + spec: MTP Support (#22673)

2026-05-16 20:06:23 +08:00

qwen3vl.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

qwen3vlmoe.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

qwen35.cpp

llama: add llm_graph_input_mtp (#23643)

2026-05-29 09:17:32 +03:00

qwen35moe.cpp

llama: add llm_graph_input_mtp (#23643)

2026-05-29 09:17:32 +03:00

qwen.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

refact.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

rnd1.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

rwkv6-base.cpp

models : deduplicate delta-net graphs for Qwen family (#19597)

2026-02-16 14:35:04 +02:00

rwkv6.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

rwkv6qwen2.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

rwkv7-base.cpp

models : deduplicate delta-net graphs for Qwen family (#19597)

2026-02-16 14:35:04 +02:00

rwkv7.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

seed-oss.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

smallthinker.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

smollm3.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

stablelm.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

starcoder2.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

starcoder.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

step35.cpp

StepFun 3.5 MTP (#23274)

2026-06-02 17:44:35 +02:00

t5.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

t5encoder.cpp

model: move load_hparams and load_tensors to per-model definition (#22004)

2026-05-04 12:36:59 +02:00

talkie.cpp

model : add support for talkie-1930-13b (#22596)

2026-05-26 07:57:38 +03:00

wavtokenizer-dec.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00

xverse.cpp

model : NvFP4 quantized LM head support (#23046)

2026-05-16 11:09:27 +02:00