apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-06 17:17:14 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix enable_logprob not in rl_config (#2808)

Browse Source
This commit is contained in:
chen
2025-07-11 11:52:48 +08:00
committed by GitHub
parent 4025ea7e5b
commit 94691bcd90
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
fastdeploy/rl/rollout_config.py
View File

@@ -58,6 +58,7 @@ class RolloutModelConfig:
max_capture_batch_size: int = 64, max_capture_batch_size: int = 64,
guided_decoding_backend: str = "off", guided_decoding_backend: str = "off",
disable_any_whitespace: bool = True, disable_any_whitespace: bool = True,
enable_logprob: bool = False,
): ):
# Required parameters # Required parameters
self.model_name_or_path = model_name_or_path self.model_name_or_path = model_name_or_path
@@ -99,6 +100,7 @@ class RolloutModelConfig:
self.max_capture_batch_size = max_capture_batch_size self.max_capture_batch_size = max_capture_batch_size
self.guided_decoding_backend = guided_decoding_backend self.guided_decoding_backend = guided_decoding_backend
self.disable_any_whitespace = disable_any_whitespace self.disable_any_whitespace = disable_any_whitespace
self.enable_logprob = enable_logprob
def __str__(self): def __str__(self):
return "\n".join(f"{k}: {v}" for k, v in self.__dict__.items()) return "\n".join(f"{k}: {v}" for k, v in self.__dict__.items())