Timothy J. Baek пре 1 година
родитељ
комит
7e0a26ef4e
1 измењених фајлова са 6 додато и 1 уклоњено
  1. 6 1
      backend/open_webui/apps/rag/main.py

+ 6 - 1
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