From 9797394c81133eebb367bd0673b6c89eefd5a38e Mon Sep 17 00:00:00 2001
From: Mug <2797716+SagsMug@users.noreply.github.com>
Date: Sat, 6 May 2023 13:27:52 +0200
Subject: [PATCH] Wrong logit_bias parsed type

---
 examples/low_level_api/common.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/low_level_api/common.py b/examples/low_level_api/common.py
index 7a25582..2bfe356 100644
--- a/examples/low_level_api/common.py
+++ b/examples/low_level_api/common.py
@@ -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