TensorRT-LLMs/tests/unittest/api_stability/references/sampling_params.yaml
Robin Kobus db8c63b9b1
[TRTLLM-4517] [feat] Additional model outputs (#7206)
Signed-off-by: Robin Kobus <19427718+Funatiq@users.noreply.github.com>
2025-10-13 15:33:18 +02:00

17 lines
391 B
YAML

methods:
__init__:
parameters:
# Experimental features
additional_model_outputs:
annotation: Optional[List[str]]
default: null
status: prototype
return_perf_metrics:
annotation: bool
default: false
beam_width_array:
annotation: Optional[List[int]]
default: null
return_annotation: None
properties: {}