vllm/vllm/model_executor/models at cc7ae5e7cab77765369630c1401410ca54184065 - vllm

__init__.py

[Core] Move multimodal placeholder from chat utils to model definition (#20355 )

2025-07-03 08:18:30 +00:00

adapters.py

[Model] Pooling models default to using chunked prefill & prefix caching if supported. (#20930 )

2025-08-11 09:41:37 -07:00

aimv2.py

[Misc] Use config definitions from Transformers library (#21913 )

2025-08-08 23:10:51 -07:00

arcee.py

Fix Arcee model weight loading: Add custom load_weights (#21725 )

2025-08-04 04:09:56 -07:00

arctic.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

aria.py

[Refactor] Define MultiModalKwargsItems separate from MultiModalKwargs (#23053 )

2025-08-18 09:52:00 +00:00

aya_vision.py

[Bugfix] Ensure correctness of Cohere2Vision processing (#23245 )

2025-08-20 11:09:18 +00:00

baichuan.py

[CI] change spell checker from codespell to typos (#18711 )

2025-06-11 19:57:10 -07:00

bailing_moe.py

support torch.compile for bailing moe (#21664 )

2025-07-26 23:54:32 +00:00

bamba.py

[V1] [Hybrid] Support using float32 for state in Hybrid Models (Mamba2, Mamba1, Minimax) (#22928 )

2025-08-15 12:57:06 +00:00

bart.py

[New Model]mBART model (#22883 )

2025-08-16 12:16:58 +00:00

bert_with_rope.py

[Attention] Refactor AttentionMetadata Preparation for Encoder-only Models (#23154 )

2025-08-22 05:05:59 +00:00

bert.py

[Attention] Refactor AttentionMetadata Preparation for Encoder-only Models (#23154 )

2025-08-22 05:05:59 +00:00

blip2.py

[Refactor] Define MultiModalKwargsItems separate from MultiModalKwargs (#23053 )

2025-08-18 09:52:00 +00:00

blip.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

bloom.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

chameleon.py

[Refactor] Define MultiModalKwargsItems separate from MultiModalKwargs (#23053 )

2025-08-18 09:52:00 +00:00

chatglm.py

[Misc] Modify the organization of GLM series (#22171 )

2025-08-03 23:51:20 -07:00

clip.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

cohere2_vision.py

[CI/Build] Split out mm processor tests (#23260 )

2025-08-20 20:05:20 -07:00

commandr.py

Refactor sliding window configuration to Transformers best practice (#21927 )

2025-08-09 20:50:48 -07:00

config.py

[V1] [Hybrid] Support using float32 for state in Hybrid Models (Mamba2, Mamba1, Minimax) (#22928 )

2025-08-15 12:57:06 +00:00

constant_size_cache.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

dbrx.py

[Misc] Use config definitions from Transformers library (#21913 )

2025-08-08 23:10:51 -07:00

deepseek_eagle.py

[Model] Support deepseek with eagle (#21086 )

2025-08-20 19:01:31 +08:00

deepseek_mtp.py

[Bugfix] Fix DeepSeek MTP (#22934 )

2025-08-16 01:25:06 +00:00

deepseek_v2.py

[Feature] use --eplb_config to set eplb param (#20562 )

2025-08-20 14:07:28 -07:00

deepseek_vl2.py

[Multimodal] Always enable hashing mm data (#23308 )

2025-08-21 07:23:28 -07:00

deepseek.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

dots1.py

[Misc] Use config definitions from Transformers library (#21913 )

2025-08-08 23:10:51 -07:00

ernie45_moe.py

[Bugfix] Fix Ernie4_5_MoeForCausalLM shared experts (#21717 )

2025-07-28 11:02:25 +00:00

ernie45.py

[Ernie 4.5] Name Change for Base 0.3B Model (#21735 )

2025-07-28 12:22:32 +00:00

ernie_mtp.py

[Model][V1] Support Ernie MTP (#22169 )

2025-08-20 20:41:55 +08:00

exaone4.py

Refactor sliding window configuration to Transformers best practice (#21927 )

2025-08-09 20:50:48 -07:00

exaone.py

[Misc] Remove redundant config definitions (#21891 )

2025-07-29 23:54:18 -07:00

fairseq2_llama.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

falcon_h1.py

[V1] [Hybrid] Support using float32 for state in Hybrid Models (Mamba2, Mamba1, Minimax) (#22928 )

2025-08-15 12:57:06 +00:00

falcon.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

florence2.py

[Bugfix] Fix broken Florence-2 model (#23426 )

2025-08-22 17:50:52 +00:00

fuyu.py

[Refactor] Define MultiModalKwargsItems separate from MultiModalKwargs (#23053 )

2025-08-18 09:52:00 +00:00

gemma2.py

Refactor sliding window configuration to Transformers best practice (#21927 )

2025-08-09 20:50:48 -07:00

gemma3_mm.py

[Refactor] Define MultiModalKwargsItems separate from MultiModalKwargs (#23053 )

2025-08-18 09:52:00 +00:00

gemma3.py

Refactor sliding window configuration to Transformers best practice (#21927 )

2025-08-09 20:50:48 -07:00

gemma3n_mm.py

[Refactor] Define MultiModalKwargsItems separate from MultiModalKwargs (#23053 )

2025-08-18 09:52:00 +00:00

gemma3n.py

Refactor sliding window configuration to Transformers best practice (#21927 )

2025-08-09 20:50:48 -07:00

gemma.py

[Model] Re-add the implicit conversion feature for as_seq_cls_model (#21103 )

2025-07-18 07:15:07 +00:00

glm4_1v.py

[BugFix] Fix the issue where image embeddings were incorrectly split.… (#23366 )

2025-08-22 16:56:46 +00:00

glm4_moe_mtp.py

[Bugfix] Fix DeepSeek MTP (#22934 )

2025-08-16 01:25:06 +00:00

glm4_moe.py

[Feature] use --eplb_config to set eplb param (#20562 )

2025-08-20 14:07:28 -07:00

glm4.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

glm4v.py

[Refactor] Define MultiModalKwargsItems separate from MultiModalKwargs (#23053 )

2025-08-18 09:52:00 +00:00

glm.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

gpt2.py

[Model][1/N] Support multiple poolers at model level (#21227 )

2025-07-21 02:22:21 -07:00

gpt_bigcode.py

Fix loading of quantized BigCode models (#22463 )

2025-08-08 23:12:12 -07:00

gpt_j.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

gpt_neox.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

gpt_oss.py

[Model] use autoWeightsLoader for gptoss (#22446 )

2025-08-20 10:16:27 +00:00

granite_speech.py

[Refactor] Define MultiModalKwargsItems separate from MultiModalKwargs (#23053 )

2025-08-18 09:52:00 +00:00

granite.py

[Bugfix] Fix Tensor Parallelism Padding Consistency in Granite Models (#20843 )

2025-07-12 06:11:30 -07:00

granitemoe.py

Let GraniteMoeAttention use YaRN (#21174 )

2025-07-18 05:52:52 -07:00

granitemoehybrid.py

[Model] Granite-4 support loading quantized checkpoint (#22925 )

2025-08-15 18:47:56 +00:00

granitemoeshared.py

Let GraniteMoeAttention use YaRN (#21174 )

2025-07-18 05:52:52 -07:00

gritlm.py

[Doc] Update V1 status of various pooling models (#23189 )

2025-08-20 10:33:41 +08:00

grok1.py

[Quantization] Enable BNB support for more MoE models (#21100 )

2025-07-18 17:52:02 -07:00

h2ovl.py

[Multimodal] Always enable hashing mm data (#23308 )

2025-08-21 07:23:28 -07:00

hunyuan_v1.py

[Model] add Hunyuan V1 Dense Model support. (#21368 )

2025-07-23 03:54:08 -07:00

hyperclovax_vision.py

[Bugfix] Fix extra whitespace in strings caused by newline (#23272 )

2025-08-20 22:03:00 -07:00

idefics2_vision_model.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

idefics3.py

[Refactor] Define MultiModalKwargsItems separate from MultiModalKwargs (#23053 )

2025-08-18 09:52:00 +00:00

interfaces_base.py

Update transformers to v4.55 (#21931 )

2025-08-05 22:56:14 -07:00

interfaces.py

[Doc] Update V1 status of various pooling models (#23189 )

2025-08-20 10:33:41 +08:00

intern_vit.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

internlm2_ve.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

internlm2.py

[Model] Pooling models default to using chunked prefill & prefix caching if supported. (#20930 )

2025-08-11 09:41:37 -07:00

interns1_vit.py

Support Intern-S1 (#21628 )

2025-07-26 19:14:04 +08:00

interns1.py

[Refactor] Define MultiModalKwargsItems separate from MultiModalKwargs (#23053 )

2025-08-18 09:52:00 +00:00

internvl.py

[Refactor] Define MultiModalKwargsItems separate from MultiModalKwargs (#23053 )

2025-08-18 09:52:00 +00:00

jais.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

jamba.py

[V1][Mamba1] - Full CUDA and Piecewise CUDA Graphs Support (#23035 )

2025-08-20 20:08:51 -07:00

jina_vl.py

[Model] Pooling model activation supports per request control by PoolingParams (#20538 )

2025-08-05 00:37:00 -07:00

keye.py

[Refactor] Define MultiModalKwargsItems separate from MultiModalKwargs (#23053 )

2025-08-18 09:52:00 +00:00

kimi_vl.py

[Model] Support Pipeline Parallelism for moonshotai/Kimi-VL-A3B-Thinking-2506 (#23114 )

2025-08-19 14:24:31 +08:00

lfm2.py

[Model] Add LFM2 architecture (#22845 )

2025-08-21 09:35:07 +02:00

llama4_eagle.py

[Meta] Official Eagle mm support, first enablement on llama4 (#20788 )

2025-07-31 10:35:07 -07:00

llama4.py

[bug fix] Fix llama4 spec decoding (#22691 )

2025-08-19 08:53:24 +00:00

llama_eagle3.py

[Speculative Decoding] Add speculators config support (#21345 )

2025-08-01 08:25:18 -04:00

llama_eagle.py

[Meta] Official Eagle mm support, first enablement on llama4 (#20788 )

2025-07-31 10:35:07 -07:00

llama.py

[Attention] Refactor AttentionMetadata Preparation for Encoder-only Models (#23154 )

2025-08-22 05:05:59 +00:00

llava_next_video.py

[Refactor] Define MultiModalKwargsItems separate from MultiModalKwargs (#23053 )

2025-08-18 09:52:00 +00:00

llava_next.py

Migrate LlavaNextImageInputs to TensorSchema (#21774 )

2025-08-10 09:05:21 -07:00

llava_onevision.py

Migrate LlavaOnevisionMultiInputs to TensorSchema (#21844 )

2025-08-19 17:02:02 +00:00

llava.py

[Multimodal] Always enable hashing mm data (#23308 )

2025-08-21 07:23:28 -07:00

mamba2.py

[V1] [Hybrid] Support using float32 for state in Hybrid Models (Mamba2, Mamba1, Minimax) (#22928 )

2025-08-15 12:57:06 +00:00

mamba_cache.py

[V1] [Hybrid] Support using float32 for state in Hybrid Models (Mamba2, Mamba1, Minimax) (#22928 )

2025-08-15 12:57:06 +00:00

mamba.py

[V1][Mamba1] - Full CUDA and Piecewise CUDA Graphs Support (#23035 )

2025-08-20 20:08:51 -07:00

medusa.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

mimo_mtp.py

[Bugfix] Fix DeepSeek MTP (#22934 )

2025-08-16 01:25:06 +00:00

mimo.py

[Model] Remove model sampler (#21059 )

2025-07-16 19:03:37 +00:00

minicpm3.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

minicpm_eagle.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

minicpm.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

minicpmo.py

Migrate MiniCPMOAudioInputs to TensorSchema (#21847 )

2025-08-22 16:43:29 +08:00

minicpmv.py

[Refactor] Define MultiModalKwargsItems separate from MultiModalKwargs (#23053 )

2025-08-18 09:52:00 +00:00

minimax_cache.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

minimax_text_01.py

[V1] [Hybrid] Support using float32 for state in Hybrid Models (Mamba2, Mamba1, Minimax) (#22928 )

2025-08-15 12:57:06 +00:00

minimax_vl_01.py

[Bugfix] Fix broken Minimax-01-VL model (#22116 )

2025-08-19 08:49:29 +00:00

mistral3.py

Migrate Mistral3ImagePixelInputs to TensorSchema (#21945 )

2025-08-20 17:37:29 +00:00

mixtral_quant.py

[CI] change spell checker from codespell to typos (#18711 )

2025-06-11 19:57:10 -07:00

mixtral.py

[Quantization] add BNB for MixtralForCausalLM (#20893 )

2025-07-14 07:34:34 +00:00

mllama4.py

[CLI][Doc] Formalize --mm-encoder-tp-mode (#23190 )

2025-08-20 23:42:28 +08:00

mllama.py

Migrate MllamaImagePixelInputs to TensorSchema (#22020 )

2025-08-22 11:28:49 +08:00

mlp_speculator.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

modernbert.py

[Attention] Refactor AttentionMetadata Preparation for Encoder-only Models (#23154 )

2025-08-22 05:05:59 +00:00

module_mapping.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

molmo.py

Migrate MolmoImageInputs to TensorSchema (#22022 )

2025-08-21 16:54:08 +00:00

moonvit.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

mpt.py

[CI/Build] Fix registry tests (#21934 )

2025-07-30 09:10:41 -07:00

nemotron_h.py

[V1] [Hybrid] Support using float32 for state in Hybrid Models (Mamba2, Mamba1, Minimax) (#22928 )

2025-08-15 12:57:06 +00:00

nemotron_nas.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

nemotron_vl.py

Migrate InternVLImagePixelInputs (in nemotron_vl.py) to TensorSchema (#22023 )

2025-08-19 13:48:26 +08:00

nemotron.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

nvlm_d.py

[Refactor] Define MultiModalKwargsItems separate from MultiModalKwargs (#23053 )

2025-08-18 09:52:00 +00:00

olmo2.py

[Model] Improve olmo and olmo2 (#23228 )

2025-08-20 12:47:05 +00:00

olmo.py

[Model] Improve olmo and olmo2 (#23228 )

2025-08-20 12:47:05 +00:00

olmoe.py

[Misc] Use config definitions from Transformers library (#21913 )

2025-08-08 23:10:51 -07:00

opt.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

orion.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

ovis2_5.py

[Model] Add Ovis2.5 PP support (#23405 )

2025-08-22 17:46:34 +00:00

ovis.py

[Refactor] Define MultiModalKwargsItems separate from MultiModalKwargs (#23053 )

2025-08-18 09:52:00 +00:00

paligemma.py

[Multimodal] Always enable hashing mm data (#23308 )

2025-08-21 07:23:28 -07:00

persimmon.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

phi3.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

phi3v.py

[Refactor] Define MultiModalKwargsItems separate from MultiModalKwargs (#23053 )

2025-08-18 09:52:00 +00:00

phi4_multimodal.py

[Refactor] Define MultiModalKwargsItems separate from MultiModalKwargs (#23053 )

2025-08-18 09:52:00 +00:00

phi4flash.py

[BugFix] Fix regression caused by mamba state dtype PR (#22998 )

2025-08-15 22:55:26 +00:00

phi4mm_audio.py

[CI] change spell checker from codespell to typos (#18711 )

2025-06-11 19:57:10 -07:00

phi4mm_utils.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

phi4mm.py

[Bugfix] Fix extra whitespace in strings caused by newline (#23272 )

2025-08-20 22:03:00 -07:00

phi.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

phimoe.py

[Misc] Fix PhiMoE expert mapping (#21085 )

2025-07-17 05:47:49 +00:00

pixtral.py

[Multimodal] Always enable hashing mm data (#23308 )

2025-08-21 07:23:28 -07:00

plamo2.py

[BugFix] Fix regression caused by mamba state dtype PR (#22998 )

2025-08-15 22:55:26 +00:00

prithvi_geospatial_mae.py

[Multimodal] Always enable hashing mm data (#23308 )

2025-08-21 07:23:28 -07:00

qwen2_5_omni_thinker.py

[BugFix] Fix the issue where image embeddings were incorrectly split.… (#23366 )

2025-08-22 16:56:46 +00:00

qwen2_5_vl.py

[CLI][Doc] Formalize --mm-encoder-tp-mode (#23190 )

2025-08-20 23:42:28 +08:00

qwen2_audio.py

[Refactor] Define MultiModalKwargsItems separate from MultiModalKwargs (#23053 )

2025-08-18 09:52:00 +00:00

qwen2_moe.py

[Misc] Use config definitions from Transformers library (#21913 )

2025-08-08 23:10:51 -07:00

qwen2_rm.py

[Model] Pooling models default to using chunked prefill & prefix caching if supported. (#20930 )

2025-08-11 09:41:37 -07:00

qwen2_vl.py

[BugFix] Fix the issue where image embeddings were incorrectly split.… (#23366 )

2025-08-22 16:56:46 +00:00

qwen2.py

[Speculators][Speculative Decoding] Fix Qwen 2 Eagle3 Support (#23337 )

2025-08-22 16:39:19 +00:00

qwen3_moe.py

[Feature] use --eplb_config to set eplb param (#20562 )

2025-08-20 14:07:28 -07:00

qwen3.py

Add: SupportsEagle3 interface for explicit EAGLE3 support (#22642 )

2025-08-12 09:24:52 -07:00

qwen_vl.py

[Refactor] Define MultiModalKwargsItems separate from MultiModalKwargs (#23053 )

2025-08-18 09:52:00 +00:00

qwen.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

registry.py

[New Model] Add Seed-Oss model (#23241 )

2025-08-22 04:58:10 +00:00

roberta.py

[Performance] V1 Pooling Models E2E Performance Optimization (#23162 )

2025-08-21 13:26:09 +00:00

rvl.py

[Model][VLM] Support R-4B Model (#23246 )

2025-08-21 04:08:52 +00:00

seed_oss.py

[New Model] Add Seed-Oss model (#23241 )

2025-08-22 04:58:10 +00:00

siglip2navit.py

[Model] Add Ovis2.5 PP support (#23405 )

2025-08-22 17:46:34 +00:00

siglip.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

skyworkr1v.py

[Refactor] Define MultiModalKwargsItems separate from MultiModalKwargs (#23053 )

2025-08-18 09:52:00 +00:00

smolvlm.py

[Misc] Automatically resolve HF processor init kwargs (#22005 )

2025-07-31 22:44:10 -07:00

solar.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

stablelm.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

starcoder2.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

step3_text.py

[Model] Add step3 vl (#21998 )

2025-07-31 23:19:06 +08:00

step3_vl.py

[CLI][Doc] Formalize --mm-encoder-tp-mode (#23190 )

2025-08-20 23:42:28 +08:00

tarsier.py

[Refactor] Define MultiModalKwargsItems separate from MultiModalKwargs (#23053 )

2025-08-18 09:52:00 +00:00

telechat2.py

[CI/Build] Fix registry tests (#21934 )

2025-07-30 09:10:41 -07:00

teleflm.py

[Misc] Add SPDX-FileCopyrightText (#19100 )

2025-06-03 11:20:17 -07:00

transformers.py

[Multimodal] Always enable hashing mm data (#23308 )

2025-08-21 07:23:28 -07:00

ultravox.py

[Refactor] Define MultiModalKwargsItems separate from MultiModalKwargs (#23053 )

2025-08-18 09:52:00 +00:00

utils.py

Improve fast_topk function with type hints and documentation (#22530 )

2025-08-09 20:25:42 -07:00

vision.py

[FEAT][ROCm] Enable running Flash Attention as ViT attn backend for Qwen-VL models on ROCm platform. (#22069 )

2025-08-01 23:53:18 -07:00

voxtral.py

[Multimodal] Always enable hashing mm data (#23308 )

2025-08-21 07:23:28 -07:00

whisper.py

[Refactor] Define MultiModalKwargsItems separate from MultiModalKwargs (#23053 )

2025-08-18 09:52:00 +00:00

zamba2.py

[V1] [Hybrid] Support using float32 for state in Hybrid Models (Mamba2, Mamba1, Minimax) (#22928 )

2025-08-15 12:57:06 +00:00