From 24c3f7a664bb857d2f97b0e4fcb1aa9d4250d034 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Sat, 21 Dec 2024 16:29:48 -0700 Subject: [PATCH] fix: custom model --- backend/open_webui/routers/openai.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/backend/open_webui/routers/openai.py b/backend/open_webui/routers/openai.py index 370513255..96eac81fe 100644 --- a/backend/open_webui/routers/openai.py +++ b/backend/open_webui/routers/openai.py @@ -533,6 +533,9 @@ async def generate_chat_completion( user=Depends(get_verified_user), bypass_filter: Optional[bool] = False, ): + if BYPASS_MODEL_ACCESS_CONTROL: + bypass_filter = True + idx = 0 payload = {**form_data} if "metadata" in payload: @@ -545,6 +548,7 @@ async def generate_chat_completion( if model_info: if 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() payload = apply_model_params_to_body_openai(params, payload)