mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-01-25 05:02:59 +08:00
* committed APIs validation Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com> * fix Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com> * clean name Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com> * separate Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com> * add TODOs Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com> * fix naming Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com> * fix Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com> --------- Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com>
29 lines
747 B
YAML
29 lines
747 B
YAML
methods:
|
|
__init__:
|
|
parameters:
|
|
# Experimental features
|
|
additional_model_outputs:
|
|
annotation: Optional[List[tensorrt_llm.sampling_params.AdditionalModelOutput]]
|
|
default: null
|
|
return_perf_metrics:
|
|
annotation: bool
|
|
default: false
|
|
# TODO [TRTLLM-3716]: Deprecated arguments
|
|
beam_width:
|
|
annotation: int
|
|
default: 1
|
|
max_new_tokens:
|
|
annotation: Optional[int]
|
|
default: null
|
|
min_length:
|
|
annotation: Optional[int]
|
|
default: null
|
|
num_return_sequences:
|
|
annotation: Optional[int]
|
|
default: null
|
|
random_seed:
|
|
annotation: Optional[int]
|
|
default: null
|
|
return_annotation: None
|
|
properties: {}
|