We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 74ef64e commit 267cf1fCopy full SHA for 267cf1f
megatron/core/inference/sampling_params.py
@@ -24,7 +24,7 @@ class SamplingParams:
24
return_log_probs: bool = False
25
skip_prompt_log_probs: bool = False
26
return_segments: bool = False # Whether to return individually detokenized tokens
27
- num_tokens_to_generate: int = 30
+ num_tokens_to_generate: int = None
28
num_tokens_total: Optional[int] = None # Cannot set both this and num_tokens_to_generate
29
termination_id: Optional[int] = None
30
top_n_logprobs: int = 0
0 commit comments