UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-01-18 16:51:17 +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_admin_user,
)
from utils.task import prompt_template
from config import (
SRC_LOG_LEVELS,
@ -47,7 +45,6 @@ from config import (
from utils.misc import (
apply_model_params_to_body_ollama,
calculate_sha256,
add_or_update_system_message,
apply_model_params_to_body_openai,
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 = {**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):