Skip to content

Commit 2c36074

Browse files
authored
check (#2811)
1 parent b5e4288 commit 2c36074

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

fastdeploy/rl/rollout_config.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,7 @@ def __init__(
5858
max_capture_batch_size: int = 64,
5959
guided_decoding_backend: str = "off",
6060
disable_any_whitespace: bool = True,
61+
enable_logprob: bool = False,
6162
):
6263
# Required parameters
6364
self.model_name_or_path = model_name_or_path
@@ -99,6 +100,7 @@ def __init__(
99100
self.max_capture_batch_size = max_capture_batch_size
100101
self.guided_decoding_backend = guided_decoding_backend
101102
self.disable_any_whitespace = disable_any_whitespace
103+
self.enable_logprob = enable_logprob
102104

103105
def __str__(self):
104106
return "\n".join(f"{k}: {v}" for k, v in self.__dict__.items())

0 commit comments

Comments
 (0)