This commit is contained in:
Timothy J. Baek 2024-09-17 23:07:04 +02:00
parent 65b7c9898b
commit 7e0a26ef4e
1 changed files with 6 additions and 1 deletions

View File

@ -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