Skip to content

Commit 9797394

Browse files
committed
Wrong logit_bias parsed type
1 parent 1895c11 commit 9797394

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/low_level_api/common.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -180,7 +180,7 @@ def gpt_params_parse(argv = None):
180180
if (logit_bias_str != None):
181181
for i in logit_bias_str:
182182
if (m := re.match(r"(\d+)([-+]\d+)", i)):
183-
params.logit_bias[int(m.group(1))] = int(m.group(2))
183+
params.logit_bias[int(m.group(1))] = float(m.group(2))
184184

185185
return params
186186

0 commit comments

Comments
 (0)