Files

T

425c4eafb0 [Sampler] Stop upcasting logits to fp32 in apply_sampling_params (#48641 )

Signed-off-by: mgoin <mgoin64@gmail.com>
Signed-off-by: Nick Hill <nickhill123@gmail.com>
Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
Co-authored-by: Nick Hill <nickhill123@gmail.com>

2026-07-17 18:15:20 -07:00

metrics

[Model Runner V2] Support num NaNs in logits (#30187 )

2025-12-09 10:00:49 -08:00

[Bugfix][Multimodal] Fix Qwen3-Omni use_audio_in_video with mixed image/video inputs (#46213 )

2026-07-17 08:31:16 +00:00

model_states

[Bugfix][Multimodal] Fix Qwen3-Omni use_audio_in_video with mixed image/video inputs (#46213 )

2026-07-17 08:31:16 +00:00

pool

[Core] Move MRV1 late_interaction_runner.py out of MRV2 subtree (#48014 )

2026-07-08 18:30:11 +00:00

sample

[Sampler] Stop upcasting logits to fp32 in apply_sampling_params (#48641 )

2026-07-17 18:15:20 -07:00

spec_decode

[Bugfix][V1/V2] Fix prompt_logprobs to respect logprobs_mode (#47680 )

2026-07-17 21:58:59 +01:00

__init__.py

GPU Model Runner V2 (#25266 )

2025-11-21 08:20:55 -08:00

async_utils.py

[Perf] Use blocking CUDA events to avoid busy polling cuda driver lock (#47081 )

2026-07-07 09:36:10 +00:00

attn_utils.py

[Misc] Use meta tensor for KV cache stride calculation (#47316 )

2026-07-10 23:24:59 -04:00

block_table.py

[PERF] Fuse multi-group block table staged writes (#44944 )

2026-06-16 10:53:27 -07:00

buffer_utils.py

[Core] Ensure memory is pinned prior to async h2d copy (#45424 )

2026-06-20 20:02:24 -07:00

cp_utils.py

[Model Runner V2] Minor simplification for DCP (#34786 )

2026-02-18 11:04:53 -08:00

cudagraph_utils.py

[Model] Add PW CUDA graph support for Inkling [2/N] (#48822 )

2026-07-16 10:28:56 -07:00

dp_utils.py

[Model Runner V2][Bugfix] Fix MRV2 LoRA warmup (#35536 )

2026-06-15 13:17:23 -07:00

eplb_utils.py

[EPLB] Mask padding in EPLB load recording (#38128 )

2026-06-28 19:43:58 -07:00

input_batch.py

[Bugfix] Fix PD disagg + MTP correctness for Qwen3.5(GDN) (#47466 )

2026-07-07 13:51:22 +00:00

kv_connector.py

[Core] Cleanup KVConnector handling with PP + fix MRV2 (#43732 )

2026-05-28 13:12:03 -07:00

lora_utils.py

[Model Runner V2][Bugfix] Fix MRV2 LoRA warmup (#35536 )

2026-06-15 13:17:23 -07:00

model_runner.py

[Bugfix][V1/V2] Fix prompt_logprobs to respect logprobs_mode (#47680 )

2026-07-17 21:58:59 +01:00

pp_utils.py

[XPU] Fix PP accuracy on XPU device (#47253 )

2026-07-07 09:17:27 +08:00

README.md

GPU Model Runner V2 (#25266 )

2025-11-21 08:20:55 -08:00

shutdown.py

[MRV2] Add shutdown() method (#41297 )

2026-05-02 21:06:30 -07:00

states.py

[Bugfix] Fix PD disagg + MTP correctness for Qwen3.5(GDN) (#47466 )

2026-07-07 13:51:22 +00:00

structured_outputs.py

[ModelRunner V2] Misc minor simplifications and optimizations (#33467 )

2026-02-01 22:17:14 +00:00

warmup.py

[ModelRunner V2] Warmup cross-attn properly in encoder-decoder case (#47308 )

2026-07-01 12:36:48 -07:00

README.md

[Experimental] Model Runner V2

This directory contains the new model runner which is under active development. Ping Woosuk Kwon for any changes.