From 7e0a26ef4e1655569be5e41af79d64246cb9104f Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Tue, 17 Sep 2024 23:07:04 +0200 Subject: [PATCH] refac --- backend/open_webui/apps/rag/main.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/backend/open_webui/apps/rag/main.py b/backend/open_webui/apps/rag/main.py index f6b57fe31..9ce70f324 100644 --- a/backend/open_webui/apps/rag/main.py +++ b/backend/open_webui/apps/rag/main.py @@ -265,7 +265,12 @@ def update_reranking_model( return scores - app.state.sentence_transformer_rf = ColBERT(reranking_model) + try: + app.state.sentence_transformer_rf = ColBERT(reranking_model) + except: + log.error("ColBERT error") + app.state.sentence_transformer_rf = None + app.state.config.ENABLE_RAG_HYBRID_SEARCH = False else: import sentence_transformers