ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-12 20:35:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

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}")