We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b5e4288 commit 2c36074Copy full SHA for 2c36074
fastdeploy/rl/rollout_config.py
@@ -58,6 +58,7 @@ def __init__(
58
max_capture_batch_size: int = 64,
59
guided_decoding_backend: str = "off",
60
disable_any_whitespace: bool = True,
61
+ enable_logprob: bool = False,
62
):
63
# Required parameters
64
self.model_name_or_path = model_name_or_path
@@ -99,6 +100,7 @@ def __init__(
99
100
self.max_capture_batch_size = max_capture_batch_size
101
self.guided_decoding_backend = guided_decoding_backend
102
self.disable_any_whitespace = disable_any_whitespace
103
+ self.enable_logprob = enable_logprob
104
105
def __str__(self):
106
return "\n".join(f"{k}: {v}" for k, v in self.__dict__.items())
0 commit comments