From b7f075948529ee8b7af4002556806d2db724302c Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Mon, 16 Sep 2024 12:36:43 +0200 Subject: [PATCH] refac --- backend/open_webui/apps/rag/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/open_webui/apps/rag/main.py b/backend/open_webui/apps/rag/main.py index eab197273..423c1ab3c 100644 --- a/backend/open_webui/apps/rag/main.py +++ b/backend/open_webui/apps/rag/main.py @@ -199,7 +199,7 @@ def update_reranking_model( update_model: bool = False, ): if reranking_model: - if reranking_model in ["jinaai/jina-colbert-v2"]: + if any(model in reranking_model for model in ["jinaai/jina-colbert-v2"]): class Colbert: def __init__(self, name) -> None: