Giancarlo Delfin 5daf62271d [Model Runner V2] Fuse probabilistic rejection sample kernels (#38496) ...

Signed-off-by: Giancarlo Delfin <gdelfin@inferact.ai>

2026-04-07 17:37:37 -07:00

..

__init__.py

[Model Runner V2] Add sample/ directory and reorganize files (#29719)

2025-11-29 00:41:01 -08:00

bad_words.py

[ModelRunnerV2] Rename sampler functions and variables for clarity (#35459)

2026-03-02 19:48:56 -08:00

gumbel.py

[Model Runner V2] Fuse probabilistic rejection sample kernels (#38496)

2026-04-07 17:37:37 -07:00

logit_bias.py

[ModelRunnerV2] Rename sampler functions and variables for clarity (#35459)

2026-03-02 19:48:56 -08:00

logprob.py

[ModelRunner V2] Revert token rank comparison difference for now (#34017)

2026-02-07 11:11:05 +08:00

min_p.py

[ModelRunnerV2] Rename sampler functions and variables for clarity (#35459)

2026-03-02 19:48:56 -08:00

output.py

[Model Runner V2] Add probabilistic rejection sampling for spec decoding (#35461)

2026-03-11 14:04:32 -07:00

penalties.py

[ModelRunnerV2] Rename sampler functions and variables for clarity (#35459)

2026-03-02 19:48:56 -08:00

prompt_logprob.py

[Model Runner V2] support bad_words sampling param (#33433)

2026-02-16 16:36:06 -08:00

sampler.py

[Model Runner V2] Fix processed logits in sample() (#37144)

2026-03-16 00:35:49 -07:00

states.py

[ModelRunnerV2] Rename sampler functions and variables for clarity (#35459)

2026-03-02 19:48:56 -08:00