Skip to content

Commit a014751

Browse files
authored
Support min_tokens parameter (#37)
1 parent 657954c commit a014751

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -205,7 +205,7 @@ def get_sampling_params_dict(self, params_json):
205205
if k in params_dict:
206206
params_dict[k] = float(params_dict[k])
207207

208-
int_keys = ["best_of", "max_tokens", "n", "top_k"]
208+
int_keys = ["best_of", "max_tokens", "min_tokens", "n", "top_k"]
209209
for k in int_keys:
210210
if k in params_dict:
211211
params_dict[k] = int(params_dict[k])

0 commit comments

Comments
 (0)