Skip to content

Commit 1daa2a0

Browse files
committed
Relax upper bound for num_candidates validation
Changed num_candidates validation to only enforce a minimum of 2, removing the previous maximum limit of 16. This allows for generating more than 16 candidates if desired.
1 parent 0dc0ed6 commit 1daa2a0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

optillm/plugins/genselect_plugin.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -165,8 +165,8 @@ def run(
165165
include_reasoning = config.get('include_reasoning', DEFAULT_INCLUDE_REASONING)
166166
max_tokens = config.get('max_tokens', 4096)
167167

168-
# Validate num_candidates is in reasonable range (2-16 as per paper)
169-
num_candidates = max(2, min(16, num_candidates))
168+
# Validate num_candidates is at least 2
169+
num_candidates = max(2, num_candidates)
170170

171171
logger.info(f"Generating {num_candidates} candidates with temperature={temperature}")
172172

0 commit comments

Comments
 (0)