OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-05-28 17:20:53 +00:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

fix: custom model

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2024-12-21 16:29:48 -07:00
parent 06a692282b
commit 24c3f7a664
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend/open_webui/routers/openai.py
View File

@ -533,6 +533,9 @@ async def generate_chat_completion(
user=Depends(get_verified_user), user=Depends(get_verified_user),
bypass_filter: Optional[bool] = False, bypass_filter: Optional[bool] = False,
): ):
if BYPASS_MODEL_ACCESS_CONTROL:
bypass_filter = True
idx = 0 idx = 0
payload = {**form_data} payload = {**form_data}
if "metadata" in payload: if "metadata" in payload:
@ -545,6 +548,7 @@ async def generate_chat_completion(
if model_info: if model_info:
if model_info.base_model_id: if model_info.base_model_id:
payload["model"] = model_info.base_model_id payload["model"] = model_info.base_model_id
model_id = model_info.base_model_id
params = model_info.params.model_dump() params = model_info.params.model_dump()
payload = apply_model_params_to_body_openai(params, payload) payload = apply_model_params_to_body_openai(params, payload)