From 8206c47a47908ad28b238ccf3fbfd7aac77de8f2 Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Sun, 29 Sep 2024 23:20:37 +0200 Subject: [PATCH] refac --- backend/open_webui/apps/retrieval/main.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/backend/open_webui/apps/retrieval/main.py b/backend/open_webui/apps/retrieval/main.py index 118a3e203..2a79ac90f 100644 --- a/backend/open_webui/apps/retrieval/main.py +++ b/backend/open_webui/apps/retrieval/main.py @@ -193,7 +193,8 @@ def update_reranking_model( if any(model in reranking_model for model in ["jinaai/jina-colbert-v2"]): try: app.state.sentence_transformer_rf = ColBERT( - get_model_path(reranking_model, auto_update) + get_model_path(reranking_model, auto_update), + env="docker" if DOCKER else None, ) except Exception as e: log.error(f"ColBERT: {e}")