mirror of
https://github.com/open-webui/open-webui.git
synced 2026-10-09 03:18:18 +00:00
refac
This commit is contained in:
parent
865880a0b1
commit
e34ed72e1e
1 changed files with 16 additions and 12 deletions
|
|
@ -762,18 +762,22 @@ def parse_ollama_modelfile(model_text):
|
|||
return data
|
||||
|
||||
|
||||
def convert_logit_bias_input_to_json(user_input) -> Optional[str]:
|
||||
if user_input:
|
||||
logit_bias_pairs = user_input.split(",")
|
||||
logit_bias_json = {}
|
||||
for pair in logit_bias_pairs:
|
||||
token, bias = pair.split(":")
|
||||
token = str(token.strip())
|
||||
bias = int(bias.strip())
|
||||
bias = 100 if bias > 100 else -100 if bias < -100 else bias
|
||||
logit_bias_json[token] = bias
|
||||
return json.dumps(logit_bias_json)
|
||||
return None
|
||||
def convert_logit_bias_input_to_json(logit_bias_input) -> Optional[str]:
|
||||
if not logit_bias_input:
|
||||
return None
|
||||
|
||||
if isinstance(logit_bias_input, dict):
|
||||
return json.dumps(logit_bias_input)
|
||||
|
||||
logit_bias_pairs = logit_bias_input.split(",")
|
||||
logit_bias_json = {}
|
||||
for pair in logit_bias_pairs:
|
||||
token, bias = pair.split(":")
|
||||
token = str(token.strip())
|
||||
bias = int(bias.strip())
|
||||
bias = 100 if bias > 100 else -100 if bias < -100 else bias
|
||||
logit_bias_json[token] = bias
|
||||
return json.dumps(logit_bias_json)
|
||||
|
||||
|
||||
def freeze(value):
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue