UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-25 21:38:43 +00:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy J. Baek 2024-09-29 23:20:37 +02:00
parent f7aba20d79
commit 8206c47a47
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
backend/open_webui/apps/retrieval/main.py
View File

@ -193,7 +193,8 @@ def update_reranking_model(
if any(model in reranking_model for model in ["jinaai/jina-colbert-v2"]): if any(model in reranking_model for model in ["jinaai/jina-colbert-v2"]):
try: try:
app.state.sentence_transformer_rf = ColBERT( 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: except Exception as e:
log.error(f"ColBERT: {e}") log.error(f"ColBERT: {e}")