Timothy J. Baek před 1 rokem
rodič
revize
7e0a26ef4e
1 změnil soubory, kde provedl 6 přidání a 1 odebrání
  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
                     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:
         else:
             import sentence_transformers
             import sentence_transformers