chore: format

This commit is contained in:
Timothy Jaeryang Baek
2025-03-01 07:28:00 -08:00
parent d7088efe73
commit efe8c4ca69
53 changed files with 397 additions and 7 deletions

View File

@@ -452,13 +452,14 @@ def parse_ollama_modelfile(model_text):
return data
def convert_logit_bias_input_to_json(user_input):
logit_bias_pairs = user_input.split(',')
logit_bias_pairs = user_input.split(",")
logit_bias_json = {}
for pair in logit_bias_pairs:
token, bias = pair.split(':')
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 json.dumps(logit_bias_json)