UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-01-19 01:06:45 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

fix: repeat_penalty

Browse Source
This commit is contained in:
Michael Poluektov 2024-08-08 10:52:09 +01:00
parent 670f28d694
commit e6bbce439d
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
backend/apps/ollama/main.py
View File

@ -31,8 +31,6 @@ from utils.utils import (
get_verified_user, get_verified_user,
get_admin_user, get_admin_user,
) )
from utils.task import prompt_template
from config import ( from config import (
SRC_LOG_LEVELS, SRC_LOG_LEVELS,
@ -47,7 +45,6 @@ from config import (
from utils.misc import ( from utils.misc import (
apply_model_params_to_body_ollama, apply_model_params_to_body_ollama,
calculate_sha256, calculate_sha256,
add_or_update_system_message,
apply_model_params_to_body_openai, apply_model_params_to_body_openai,
apply_model_system_prompt_to_body, apply_model_system_prompt_to_body,
) )

7
backend/utils/misc.py
View File

@ -181,7 +181,12 @@ def apply_model_params_to_body_ollama(params: dict, form_data: dict) -> dict:
] ]
mappings = {i: lambda x: x for i in opts} mappings = {i: lambda x: x for i in opts}
mappings = {**mappings, **OPENAI_MAPPINGS} mappings = {**mappings, **OPENAI_MAPPINGS}
return apply_model_params_to_body(params, form_data, mappings) form_data = apply_model_params_to_body(params, form_data, mappings)
# only param that changes name
if (param := params.get("frequency_penalty", None)) is not None:
form_data["repeat_penalty"] = param
return form_data
def get_gravatar_url(email): def get_gravatar_url(email):