alihan/llama-cpp-python
1
0
Fork 0
mirror of https://github.com/abetlen/llama-cpp-python.git synced 2023-09-07 17:34:22 +03:00
Code Issues Packages Projects Releases 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
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