baalajimaestro/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Wrong logit_bias parsed type

Browse source
This commit is contained in:
Mug 2023-05-06 13:27:52 +02:00
parent 1895c11033
commit 9797394c81
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
examples/low_level_api/common.py
View file

@ -180,7 +180,7 @@ def gpt_params_parse(argv = None):
if (logit_bias_str != None):
for i in logit_bias_str:
if (m := re.match(r"(\d+)([-+]\d+)", i)):
params.logit_bias[int(m.group(1))] = int(m.group(2))
params.logit_bias[int(m.group(1))] = float(m.group(2))
return params