Skip to content

Commit 267cf1f

Browse files
authored
Update num_tokens_to_generate default for Gym (NVIDIA#3453)
1 parent 74ef64e commit 267cf1f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

megatron/core/inference/sampling_params.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ class SamplingParams:
2424
return_log_probs: bool = False
2525
skip_prompt_log_probs: bool = False
2626
return_segments: bool = False # Whether to return individually detokenized tokens
27-
num_tokens_to_generate: int = 30
27+
num_tokens_to_generate: int = None
2828
num_tokens_total: Optional[int] = None # Cannot set both this and num_tokens_to_generate
2929
termination_id: Optional[int] = None
3030
top_n_logprobs: int = 0

0 commit comments

Comments
 (0)