Tyler Michael SmithandClaude Opus 4.6 171eb482e8 [Kernel] Add pluggable NaN/Inf tensor checks to NaN detector ...

Add check_tensor() to NaNDetector for checking arbitrary tensors at
any point in the forward pass. Uses torch.isfinite() — all ops stay
on GPU, CUDA-graph compatible, writes to the same per-token flag
array as the RMSNorm kernel checks.

Any module can register checkpoints via register() and call
check_tensor(tensor, idx) in its forward. update_layer_names()
picks up _nan_detect_indices dicts for readable names.

Wire into DeepseekV2Attention to check attn_output before o_proj,
distinguishing "attention produced NaN" from "o_proj produced NaN".

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
Signed-off-by: Tyler Michael Smith <tlrmchlsmth@gmail.com>

2026-03-28 18:58:43 -04:00

..

transformers

Fix text only inputs for MRoPE models with the Transformers modelling backend (#37055)

2026-03-16 10:31:16 +00:00

__init__.py

[Chore] Remove SupportsV0Only interface and update supported models docs (#26783)

2025-10-14 04:55:10 -07:00

adapters.py

[Bugfix] Fix score layer quantization for sequence classification models - Qwen3 (VL) Reranker (#35849)

2026-03-04 20:57:20 -08:00

afmoe.py

[Refactor] Consolidate SupportsEagle (#36063)

2026-03-13 23:22:40 +00:00

aimv2.py

[MM] Align the prefix of MMEncoderAttention with Attention (#33750)

2026-02-04 04:07:30 +00:00

apertus.py

[Refactor] Consolidate SupportsEagle (#36063)

2026-03-13 23:22:40 +00:00

arcee.py

[Refactor] Consolidate SupportsEagle (#36063)

2026-03-13 23:22:40 +00:00

arctic.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

aria.py

[Refactor] Simplify dummy data generation (#35025)

2026-02-22 20:55:27 -08:00

audioflamingo3.py

[Bugfix] Fix loading Music Flamingo (#35535)

2026-03-17 05:24:40 +00:00

AXK1.py

[Model] Add huggingface skt/A.X-K1 model (#32407)

2026-02-27 09:26:02 -08:00

aya_vision.py

[Refactor] Simplify dummy data generation (#35025)

2026-02-22 20:55:27 -08:00

bagel.py

[Refactor] Simplify dummy data generation (#35025)

2026-02-22 20:55:27 -08:00

baichuan.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

bailing_moe_linear.py

[V0 Deprecation] Deprecate virtual engine (#37195)

2026-03-18 14:30:14 -07:00

bailing_moe.py

[Bugfix] Define router_logits_dtype for remaining MoE models (#33737)

2026-02-04 13:24:14 +08:00

bamba.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

bee.py

[Refactor] Simplify dummy data generation (#35025)

2026-02-22 20:55:27 -08:00

bert_with_rope.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

bert.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

blip2.py

[Refactor] Simplify dummy data generation (#35025)

2026-02-22 20:55:27 -08:00

blip.py

[MM] Align the prefix of MMEncoderAttention with Attention (#33750)

2026-02-04 04:07:30 +00:00

bloom.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

chameleon.py

[Refactor] Simplify dummy data generation (#35025)

2026-02-22 20:55:27 -08:00

chatglm.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

clip.py

Increase Flexibility for OOV Multimodal Token Handling (#34858)

2026-03-08 20:30:49 -07:00

cohere2_vision.py

[Refactor] Simplify dummy data generation (#35025)

2026-02-22 20:55:27 -08:00

cohere_asr.py

[Models] Cohere ASR (#35809)

2026-03-17 21:04:17 +00:00

colbert.py

[Model] Consolidate score logic by introduce score_type (#36479)

2026-03-10 13:32:25 +00:00

colmodernvbert.py

[Model] Consolidate score logic by introduce score_type (#36479)

2026-03-10 13:32:25 +00:00

colpali.py

[Model] Add ColPali late interaction model for multi-modal retrieval (#36818)

2026-03-13 02:18:57 +00:00

colqwen3_5.py

[Model] Add ColQwen3.5 4.5B support (#36887)

2026-03-17 21:17:02 +00:00

colqwen3.py

[Model] Consolidate score logic by introduce score_type (#36479)

2026-03-10 13:32:25 +00:00

commandr.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

config.py

[Model] Add ColQwen3.5 4.5B support (#36887)

2026-03-17 21:17:02 +00:00

dbrx.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

deepencoder2.py

[Models] Consolidate Deepseek-OCR2 processor (#33909)

2026-02-05 18:29:20 +00:00

deepencoder.py

[PluggableLayer][MM] Add PluggableLayer for RelPosAttention (#33753)

2026-03-03 19:41:27 -08:00

deepseek_eagle3.py

Kimi k2.5 MLA based eagle3 (#36361)

2026-03-11 11:36:11 -04:00

deepseek_eagle.py

[BugFix] LoRA: Support loading base_layer of experts (#31104)

2026-01-07 14:49:39 +08:00

deepseek_mtp.py

[MTP] Validate that MTP weights are actually loaded (#35548)

2026-02-28 12:27:40 +08:00

deepseek_ocr2.py

[Bugfix] Standardize custom HF Processor init (#37289)

2026-03-17 15:38:55 +00:00

deepseek_ocr.py

[Bugfix] Standardize custom HF Processor init (#37289)

2026-03-17 15:38:55 +00:00

deepseek_v2.py

[Kernel] Add pluggable NaN/Inf tensor checks to NaN detector

2026-03-28 18:58:43 -04:00

deepseek_vl2.py

[Misc] Lazy import registered processors (#36024)

2026-03-06 06:06:45 -08:00

dots1.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

dots_ocr.py

[Refactor] Simplify dummy data generation (#35025)

2026-02-22 20:55:27 -08:00

eagle2_5_vl.py

[2/3] Refactor InternVL-based processors (#37324)

2026-03-18 22:22:19 +08:00

ernie45_moe.py

Remove padding_index from models that don't use it for better Transformers v5 compatibility (#35189)

2026-02-24 08:04:46 -08:00

ernie45_vl_moe.py

[Bugfix][CI] fix typos (#34934)

2026-03-05 17:05:46 +00:00

ernie45_vl.py

Increase Flexibility for OOV Multimodal Token Handling (#34858)

2026-03-08 20:30:49 -07:00

ernie45.py

[Model] Add PaddleOCR-VL Model Support (#27758)

2025-11-03 19:04:22 +08:00

ernie_mtp.py

[Chore] Update type annotation of input_ids in model forward (#33063)

2026-01-26 06:02:10 -08:00

ernie.py

[Model] Add support for BERT-like Chinese ERNIE pooling models (#36385)

2026-03-13 03:23:53 +00:00

exaone4.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

exaone_moe_mtp.py

Add K-EXAONE-236B-A23B (#31621)

2026-01-12 16:30:50 +00:00

exaone_moe.py

[Chore] Update type annotation of input_ids in model forward (#33063)

2026-01-26 06:02:10 -08:00

exaone.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

extract_hidden_states.py

[V0 Deprecation] Deprecate virtual engine (#37195)

2026-03-18 14:30:14 -07:00

fairseq2_llama.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

falcon_h1.py

Fix quantized Falcon-H1 model loading issues (#32728)

2026-02-02 22:31:27 -08:00

falcon.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

fireredasr2.py

[Model] Remove unused handle_oov_mm_token (#37321)

2026-03-17 19:44:52 +00:00

flex_olmo.py

[Bugfix] Define router_logits_dtype for remaining MoE models (#33737)

2026-02-04 13:24:14 +08:00

funasr.py

FunASR model bugfix (#36633)

2026-03-10 08:14:21 -07:00

funaudiochat.py

[Chore] Remove debug code in model implementation (#35883)

2026-03-03 19:50:58 -08:00

fuyu.py

[Refactor] Simplify dummy data generation (#35025)

2026-02-22 20:55:27 -08:00

gemma2.py

Make Gemma and Gemma 2 accept inputs_embeds like Gemma 3 (#36787)

2026-03-11 18:12:43 +00:00

gemma3_mm.py

Increase Flexibility for OOV Multimodal Token Handling (#34858)

2026-03-08 20:30:49 -07:00

gemma3.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

gemma3n_audio_utils.py

[Bugfix] Fix integer overflow in Gemma3n audio processing (#31657)

2026-01-10 17:52:53 +08:00

gemma3n_mm.py

Increase Flexibility for OOV Multimodal Token Handling (#34858)

2026-03-08 20:30:49 -07:00

gemma3n.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

gemma.py

Make Gemma and Gemma 2 accept inputs_embeds like Gemma 3 (#36787)

2026-03-11 18:12:43 +00:00

glm4_1v.py

[Bugfix] Support other quantization methods in glm41v (#36321)

2026-03-11 09:48:05 +00:00

glm4_moe_lite_mtp.py

[Chore] Update type annotation of input_ids in model forward (#33063)

2026-01-26 06:02:10 -08:00

glm4_moe_lite.py

Update get_expert_mapping to include self parameter (#33525)

2026-02-02 20:29:07 +08:00

glm4_moe_mtp.py

[Chore] Update type annotation of input_ids in model forward (#33063)

2026-01-26 06:02:10 -08:00

glm4_moe.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

glm4.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

glm4v.py

[2/3] Refactor InternVL-based processors (#37324)

2026-03-18 22:22:19 +08:00

glm_ocr_mtp.py

[GLM-OCR] GLM-OCR with MTP Support (#33005)

2026-01-26 06:24:43 -08:00

glm_ocr.py

[MM] Align the prefix of MMEncoderAttention with Attention (#33750)

2026-02-04 04:07:30 +00:00

glm.py

Access partial_rotary_factor from rope_parameters (#29966)

2025-12-04 18:42:49 +00:00

glmasr_utils.py

[Refactor] Remove dead private func _fp8_perm and _extract_mask_for_item (#35068)

2026-02-23 05:05:20 -08:00

glmasr.py

[Refactor] Simplify dummy data generation (#35025)

2026-02-22 20:55:27 -08:00

gpt2.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

gpt_bigcode.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

gpt_j.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

gpt_neox.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

gpt_oss.py

[Kernel] Add gpt-oss Router GEMM kernel (#37205)

2026-03-18 08:15:56 -07:00

granite_speech.py

Increase Flexibility for OOV Multimodal Token Handling (#34858)

2026-03-08 20:30:49 -07:00

granite.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

granitemoe.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

granitemoehybrid.py

fix: Apply embedding_multiplier to inputs_embeds (#34813)

2026-02-23 00:42:46 -08:00

granitemoeshared.py

Remove padding_index from models that don't use it for better Transformers v5 compatibility (#35189)

2026-02-24 08:04:46 -08:00

gritlm.py

[Model] Standardize pooling heads (#32148)

2026-01-12 17:01:49 +00:00

grok1.py

Remove padding_index from models that don't use it for better Transformers v5 compatibility (#35189)

2026-02-24 08:04:46 -08:00

h2ovl.py

[2/3] Refactor InternVL-based processors (#37324)

2026-03-18 22:22:19 +08:00

hunyuan_v1.py

[Refactor] Consolidate SupportsEagle (#36063)

2026-03-13 23:22:40 +00:00

hunyuan_vision.py

[Refactor] Consolidate SupportsEagle (#36063)

2026-03-13 23:22:40 +00:00

hyperclovax_vision_v2.py

[Model] Add HyperCLOVAX-SEED-Think-32B vision-language model support (#31471)

2026-03-10 10:59:19 +08:00

hyperclovax_vision.py

[Model] Add HyperCLOVAX-SEED-Think-32B vision-language model support (#31471)

2026-03-10 10:59:19 +08:00

hyperclovax.py

[Model] Add HyperCLOVAX-SEED-Think-14B language model support (#37107)

2026-03-16 06:40:05 +00:00

idefics2_vision_model.py

[CI][MM] Gate vision encoder attention mask to MiniCPM only, fixing Aria regression (#36206)

2026-03-06 01:17:08 -08:00

idefics3.py

[Refactor] Simplify dummy data generation (#35025)

2026-02-22 20:55:27 -08:00

interfaces_base.py

[Model] Consolidate score logic by introduce score_type (#36479)

2026-03-10 13:32:25 +00:00

interfaces.py

[Refactor] Consolidate SupportsEagle (#36063)

2026-03-13 23:22:40 +00:00

intern_vit.py

[MM] Align the prefix of MMEncoderAttention with Attention (#33750)

2026-02-04 04:07:30 +00:00

internlm2_ve.py

[Chore] Update type annotation of input_ids in model forward (#33063)

2026-01-26 06:02:10 -08:00

internlm2.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

interns1_pro.py

[CI/Build] Enable tests for recent day-0 new models (#34585)

2026-02-15 18:17:04 -08:00

interns1_vit.py

[MM] Align the prefix of MMEncoderAttention with Attention (#33750)

2026-02-04 04:07:30 +00:00

interns1.py

Increase Flexibility for OOV Multimodal Token Handling (#34858)

2026-03-08 20:30:49 -07:00

internvl.py

[2/3] Refactor InternVL-based processors (#37324)

2026-03-18 22:22:19 +08:00

iquest_loopcoder.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

isaac.py

[Bugfix][CI] fix typos (#34934)

2026-03-05 17:05:46 +00:00

jais2.py

Remove padding_index from models that don't use it for better Transformers v5 compatibility (#35189)

2026-02-24 08:04:46 -08:00

jais.py

Make JAIS compatible with Transformers v5 (#34264)

2026-02-11 12:30:37 +00:00

jamba.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

jina_vl.py

[Chore] Update type annotation of input_ids in model forward (#33063)

2026-01-26 06:02:10 -08:00

kanana_v.py

[Refactor] Simplify dummy data generation (#35025)

2026-02-22 20:55:27 -08:00

keye_vl1_5.py

[Refactor] Define MM data parser in processing info instead of processor itself (#33260)

2026-01-29 13:55:17 +08:00

keye.py

[Bugfix][CI] fix typos (#34934)

2026-03-05 17:05:46 +00:00

kimi_audio.py

[Model] Remove unused handle_oov_mm_token (#37321)

2026-03-17 19:44:52 +00:00

kimi_k25_vit.py

[Bugfix] Add quant_config in ViT of Kimi-K2.5 (#34501)

2026-02-13 16:05:34 +00:00

kimi_k25.py

Kimi k2.5 MLA based eagle3 (#36361)

2026-03-11 11:36:11 -04:00

kimi_linear.py

standardize load_weights using AutoWeightsLoader for kimi_linear and minimax_text_01 (#37371)

2026-03-18 15:05:37 +00:00

kimi_vl.py

[Refactor] Simplify dummy data generation (#35025)

2026-02-22 20:55:27 -08:00

lfm2_moe.py

[Models] LFM2: Support LoRA (#34921)

2026-02-19 22:07:23 -08:00

lfm2_siglip2.py

[BE] Rename should_torch_compile_mm_vit to should_torch_compile_mm_encoder (#36281)

2026-03-09 18:22:05 +00:00

lfm2_vl.py

fix: align lfm2 thumbnail token counting with HF (#36707)

2026-03-11 10:28:38 -07:00

lfm2.py

[Models] LFM2: Support LoRA (#34921)

2026-02-19 22:07:23 -08:00

lightonocr.py

[Models]: Make Multimodal config implicit in ViT implementation (#31972)

2026-01-24 20:34:26 +08:00

llama4_eagle.py

[Spec Decode] Reduce TP communication for speculative decoding draft token generation (#34049)

2026-02-22 14:59:16 -08:00

llama4.py

[Llama4,CI] Bring back Llama-4 bug fixes, and also fix Maverick tests (#35033)

2026-02-23 09:04:34 -05:00

llama_eagle3.py

[Speculative Decoding] Add norm_before_fc for gpt-oss draft models (#36545)

2026-03-12 23:03:32 +00:00

llama_eagle.py

Add llmcompressor fp8 kv-cache quant (per-tensor and per-attn_head) (#30141)

2026-01-22 13:29:57 -07:00

llama.py

[Refactor] Consolidate SupportsEagle (#36063)

2026-03-13 23:22:40 +00:00

llava_next_video.py

[Refactor] Simplify dummy data generation (#35025)

2026-02-22 20:55:27 -08:00

llava_next.py

Increase Flexibility for OOV Multimodal Token Handling (#34858)

2026-03-08 20:30:49 -07:00

llava_onevision.py

[Refactor] Simplify dummy data generation (#35025)

2026-02-22 20:55:27 -08:00

llava.py

[Refactor] Consolidate SupportsEagle (#36063)

2026-03-13 23:22:40 +00:00

longcat_flash_mtp.py

[Chore] Update type annotation of input_ids in model forward (#33063)

2026-01-26 06:02:10 -08:00

longcat_flash.py

[Bugfix][CI] fix typos (#34934)

2026-03-05 17:05:46 +00:00

mamba2.py

[Model][Spec Decode] Nemotron-H MTP and Mamba Speculative Decoding Support (#33726)

2026-02-24 09:49:56 -08:00

mamba.py

[Chore] Update type annotation of input_ids in model forward (#33063)

2026-01-26 06:02:10 -08:00

medusa.py

[LoRA][1/N]Remove LoRA extra vocab (#28382)

2025-11-11 11:06:21 -08:00

midashenglm.py

[Refactor] Simplify dummy data generation (#35025)

2026-02-22 20:55:27 -08:00

mimo_mtp.py

[Chore] Update type annotation of input_ids in model forward (#33063)

2026-01-26 06:02:10 -08:00

mimo_v2_flash.py

[Refactor] Consolidate SupportsEagle (#36063)

2026-03-13 23:22:40 +00:00

mimo.py

[Chore] Update type annotation of input_ids in model forward (#33063)

2026-01-26 06:02:10 -08:00

minicpm3.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

minicpm_eagle.py

[Misc][Model][Refactor] Pass the prefix into Linear layers (#31669)

2026-01-05 20:03:18 +00:00

minicpm.py

[Refactor] Consolidate SupportsEagle (#36063)

2026-03-13 23:22:40 +00:00

minicpmo.py

[Refactor] Simplify dummy data generation (#35025)

2026-02-22 20:55:27 -08:00

minicpmv.py

fix(minicpmv): fix audio inference by handling meta device in init_re… (#36751)

2026-03-11 03:06:28 -07:00

minimax_m2.py

[Bugfix] Disable TRTLLM FP8 MoE if router_logits_dtype==float32 and routing_method!=DeepSeekV3 (#33613)

2026-02-03 13:26:51 -08:00

minimax_text_01.py

standardize load_weights using AutoWeightsLoader for kimi_linear and minimax_text_01 (#37371)

2026-03-18 15:05:37 +00:00

minimax_vl_01.py

[Bugfix] Fix models and tests for transformers v5 (#33977)

2026-02-06 21:47:41 +08:00

mistral3.py

[LoRA][BugFix] Fix skipped LoRA adapters for Mistral3 (#36928)

2026-03-18 22:34:19 +00:00

mistral_large_3_eagle.py

Fix EagleMistralLarge3Model initialization (#37232)

2026-03-16 15:41:00 -07:00

mistral_large_3.py

Add Mistral Large 3 and Ministral 3 (#29757)

2025-12-02 10:29:00 +00:00

mistral.py

Remove unused logic in models/mistral.py (#33095)

2026-01-26 09:01:52 -08:00

mixtral.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

mllama4.py

[torch.compile][BE] Modify cudagraph callable to check for is_forward_context_set (#36288)

2026-03-16 19:42:34 +00:00

mlp_speculator.py

[LoRA][1/N]Remove LoRA extra vocab (#28382)

2025-11-11 11:06:21 -08:00

modernbert.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

module_mapping.py

[Misc] Remove unused ModelKeys (#32608)

2026-01-19 17:34:59 +00:00

molmo2.py

[Refactor] Remove Molmo2 processor wrapper (#36667)

2026-03-11 03:07:20 -07:00

molmo.py

[Misc] Move processors to transformers_utils (#35953)

2026-03-09 11:31:39 +08:00

moonvit.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

mpt.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

musicflamingo.py

[Bugfix] Fix loading Music Flamingo (#35535)

2026-03-17 05:24:40 +00:00

nano_nemotron_vl.py

chunk parakeet into 30s clips to prevent OOMs on long audios (#36671)

2026-03-18 14:22:24 -07:00

nemotron_h_mtp.py

[Model][Spec Decode] Nemotron-H MTP and Mamba Speculative Decoding Support (#33726)

2026-02-24 09:49:56 -08:00

nemotron_h.py

[Bugfix][CI] fix typos (#34934)

2026-03-05 17:05:46 +00:00

nemotron_nas.py

Remove padding_index from models that don't use it for better Transformers v5 compatibility (#35189)

2026-02-24 08:04:46 -08:00

nemotron_parse.py

[Model] Remove unnecessary processor definition for Nemotron Parse (#37456)

2026-03-18 18:25:13 +00:00

nemotron_vl.py

[2/3] Refactor InternVL-based processors (#37324)

2026-03-18 22:22:19 +08:00

nemotron.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

nvlm_d.py

[2/3] Refactor InternVL-based processors (#37324)

2026-03-18 22:22:19 +08:00

olmo2.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

olmo_hybrid.py

[V0 Deprecation] Deprecate virtual engine (#37195)

2026-03-18 14:30:14 -07:00

olmo.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

olmoe.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

opencua.py

[Refactor] Define MM data parser in processing info instead of processor itself (#33260)

2026-01-29 13:55:17 +08:00

openpangu_mtp.py

[Chore] Update type annotation of input_ids in model forward (#33063)

2026-01-26 06:02:10 -08:00

openpangu_vl.py

[Model] Apply #32631 for recent models (#33785)

2026-02-04 12:23:01 +00:00

openpangu.py

Remove padding_index from models that don't use it for better Transformers v5 compatibility (#35189)

2026-02-24 08:04:46 -08:00

opt.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

orion.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

ouro.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

ovis2_5.py

[Deprecation] Deprecate code in 0.17 as scheduled (#35441)

2026-02-28 17:32:37 +00:00

ovis.py

[Refactor] Simplify dummy data generation (#35025)

2026-02-22 20:55:27 -08:00

paddleocr_vl.py

[Bugfix] fix paddleocr crash on some image shape (#36959)

2026-03-13 13:46:55 +00:00

paligemma.py

[Refactor] Decouple TimingContext from InputProcessingContext (#35083)

2026-02-23 14:15:50 +00:00

parakeet.py

chunk parakeet into 30s clips to prevent OOMs on long audios (#36671)

2026-03-18 14:22:24 -07:00

persimmon.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

phi3.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

phi3v.py

Increase Flexibility for OOV Multimodal Token Handling (#34858)

2026-03-08 20:30:49 -07:00

phi4mm_audio.py

[Bugfix][CI] fix typos (#34934)

2026-03-05 17:05:46 +00:00

phi4mm_utils.py

[Bugfix][CI] fix typos (#34934)

2026-03-05 17:05:46 +00:00

phi4mm.py

[Refactor] Simplify dummy data generation (#35025)

2026-02-22 20:55:27 -08:00

phi.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

phimoe.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

pixtral.py

[Misc] Clean up renderers (#36770)

2026-03-11 16:39:29 +00:00

plamo2.py

[V0 Deprecation] Deprecate virtual engine (#37195)

2026-03-18 14:30:14 -07:00

plamo3.py

Remove padding_index from models that don't use it for better Transformers v5 compatibility (#35189)

2026-02-24 08:04:46 -08:00

qwen2_5_omni_thinker.py

[Bugfix] Fix Qwen2.5-Omni/Qwen3-Omni use_audio_in_video with multi-video inputs (#37147)

2026-03-16 08:56:06 +00:00

qwen2_5_vl.py

[torch.compile][BE] Modify cudagraph callable to check for is_forward_context_set (#36288)

2026-03-16 19:42:34 +00:00

qwen2_audio.py

[BUGFIX]Fix Qwen-Omni models audio max_token_per_item estimation error leading to encoder_cache_size is 0 (#35994)

2026-03-05 09:16:29 -08:00

qwen2_moe.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

qwen2_rm.py

[Chore] Update type annotation of input_ids in model forward (#33063)

2026-01-26 06:02:10 -08:00

qwen2_vl.py

[Bugfix][CI] fix typos (#34934)

2026-03-05 17:05:46 +00:00

qwen2.py

[Refactor] Consolidate SupportsEagle (#36063)

2026-03-13 23:22:40 +00:00

qwen3_5_mtp.py

Increase Flexibility for OOV Multimodal Token Handling (#34858)

2026-03-08 20:30:49 -07:00

qwen3_5.py

[Perf] Enable dual stream execution of input projection for Qwen3 (#36795)

2026-03-18 11:13:27 +08:00

qwen3_asr_realtime.py

[Bugfix] Add missing dynamic_arg_dims for Qwen3-ASR torch.compile (#35869)

2026-03-04 08:29:01 +00:00

qwen3_asr.py

Increase Flexibility for OOV Multimodal Token Handling (#34858)

2026-03-08 20:30:49 -07:00

qwen3_moe.py

[Refactor] Consolidate SupportsEagle (#36063)

2026-03-13 23:22:40 +00:00

qwen3_next_mtp.py

[Chore] Update type annotation of input_ids in model forward (#33063)

2026-01-26 06:02:10 -08:00

qwen3_next.py

[V0 Deprecation] Deprecate virtual engine (#37195)

2026-03-18 14:30:14 -07:00

qwen3_omni_moe_thinker.py

[Bugfix] Fix Qwen2.5-Omni/Qwen3-Omni use_audio_in_video with multi-video inputs (#37147)

2026-03-16 08:56:06 +00:00

qwen3_vl_moe.py

[Refactor] Consolidate SupportsEagle (#36063)

2026-03-13 23:22:40 +00:00

qwen3_vl.py

[Bugfix] Fix incorrect use of merge_size in Qwen3-VL video timestamp calculation (#37439)

2026-03-18 11:36:34 -07:00

qwen3.py

[Refactor] Consolidate SupportsEagle (#36063)

2026-03-13 23:22:40 +00:00

qwen_vl.py

[2/3] Refactor InternVL-based processors (#37324)

2026-03-18 22:22:19 +08:00

qwen.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

radio.py

Use MMEncoderAttention (=use FlashAttention) instead of torch.sdpa in radio.py (#35653)

2026-03-04 08:43:13 -08:00

registry.py

[Misc] Clean up model registry (#37457)

2026-03-18 18:24:44 +00:00

roberta.py

[Feature]: Support for multiple embedding types in a single inference call (#35829)

2026-03-17 17:05:42 +08:00

rvl.py

[Refactor] Simplify dummy data generation (#35025)

2026-02-22 20:55:27 -08:00

sarvam.py

Adding support to Sarvam's MoE models (#33942)

2026-03-08 01:16:24 +08:00

seed_oss.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

siglip2navit.py

[New Model] support new model ovis2.6 (#34426)

2026-02-13 00:12:45 -08:00

siglip.py

Increase Flexibility for OOV Multimodal Token Handling (#34858)

2026-03-08 20:30:49 -07:00

skyworkr1v.py

[2/3] Refactor InternVL-based processors (#37324)

2026-03-18 22:22:19 +08:00

smolvlm.py

[Bugfix] Standardize getting number of image patches/tokens (#34358)

2026-02-12 20:47:01 -08:00

solar.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

stablelm.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

starcoder2.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

step1.py

[Refactor] Consolidate SupportsEagle (#36063)

2026-03-13 23:22:40 +00:00

step3_text.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00

step3_vl.py

Increase Flexibility for OOV Multimodal Token Handling (#34858)

2026-03-08 20:30:49 -07:00

step3p5_mtp.py

[Models] Step-3.5-Flash (#33523)

2026-02-02 10:21:18 +08:00

step3p5.py

[Bugfix][CI] fix typos (#34934)

2026-03-05 17:05:46 +00:00

step_vl.py

[MM] Align the prefix of MMEncoderAttention with Attention (#33750)

2026-02-04 04:07:30 +00:00

tarsier.py

[Chore] Update type annotation of input_ids in model forward (#33063)

2026-01-26 06:02:10 -08:00

telechat2.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

teleflm.py

[LoRA][2/2]Remove LoRA extra vocab (#28545)

2025-11-21 09:46:43 +08:00

terratorch.py

Increase Flexibility for OOV Multimodal Token Handling (#34858)

2026-03-08 20:30:49 -07:00

ultravox.py

[UltraVox] Fix output type (#37224)

2026-03-17 14:51:17 +00:00

utils.py

Use Transformers v5 WeightRenaming for Transformers modeling backend (#31545)

2026-03-13 20:49:08 +00:00

vision.py

[BE] Rename should_torch_compile_mm_vit to should_torch_compile_mm_encoder (#36281)

2026-03-09 18:22:05 +00:00

voxtral_realtime.py

[Misc] Move processors to transformers_utils (#35953)

2026-03-09 11:31:39 +08:00

voxtral.py

[Misc] Clean up renderers (#36770)

2026-03-11 16:39:29 +00:00

voyage.py

Fix main pre-commit (#33975)

2026-02-06 00:08:05 -08:00

whisper_causal.py

Reapply [Attention] Refactor check_and_update_config (#35122)

2026-03-09 07:17:14 -07:00

whisper_utils.py

[Voxtral] Add new streaming arch (#32861)

2026-01-23 12:41:52 +01:00

whisper.py

Increase Flexibility for OOV Multimodal Token Handling (#34858)

2026-03-08 20:30:49 -07:00

zamba2.py

[5/N][Attention] Finish eliminating vllm/attention folder (#32064)

2026-01-27 10:02:51 -05:00