4 people 107a03ba63 [Core] Support fp32 lm_head for generation models via head_dtype (RFC #48305 §3.6) (#48390) ...

Signed-off-by: Karthik Kothuri <karthikkothuri2009@gmail.com>
Signed-off-by: wang.yuqi <yuqi.wang@daocloud.io>
Co-authored-by: Claude <noreply@anthropic.com>
Co-authored-by: wang.yuqi <yuqi.wang@daocloud.io>

2026-07-13 16:43:34 +08:00

..

__init__.py

[V1] Adding min tokens/repetition/presence/frequence penalties to V1 sampler (#10681)

2024-12-26 19:02:58 +09:00

test_batched_count_greater_than.py

[Performance Improvement] Update batched_count_greater_than to handle batch size 1 without recompile (#38933)

2026-04-09 23:51:31 +08:00

test_head_dtype.py

[Core] Support fp32 lm_head for generation models via head_dtype (RFC #48305 §3.6) (#48390)

2026-07-13 16:43:34 +08:00

test_logprobs_e2e.py

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

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

test_logprobs.py

[Bugfix] Reject sampling params unsupported by diffusion models (#45418)

2026-07-07 11:25:36 +00:00

test_rejection_sampler.py

[Spec Decode] Reject placeholder (-1) draft tokens in rejection sampler (#46533)

2026-06-24 03:32:32 +00:00

test_sampler.py

refactor hard coded device string in test files under tests/v1 and tests/lora (#37566)

2026-04-03 11:21:47 +08:00

test_sampling_params_e2e.py

[Bugfix][Model Runner V2] Preserve all allowed_token_ids in the logit bias kernel (#46245)

2026-06-23 07:01:13 +00:00

test_topk_topp_sampler.py

[Platform] Replace torch.cuda.mem_get_info with torch.accelerator.get_memory_info (#44825)

2026-06-30 14:39:52 +08:00

utils.py

[Reasoning] Support for speculative decoding with thinking budget (#34668)

2026-04-29 06:14:52 +00:00