add_thinking_budget_logits_processor#

class tensorrt_llm.llmapi.add_thinking_budget_logits_processor(
sampling_params: SamplingParams,
*,
reasoning_parser: str | None,
tokenizer: Any,
chat_template_kwargs: dict[str, Any] | None = None,
)[source]#

Bases:

Attach a thinking-budget logits processor to sampling_params.

__init__(**kwargs)#