Timothy J. Baek 1 an în urmă
părinte
comite
7e0a26ef4e
1 a modificat fișierele cu 6 adăugiri și 1 ștergeri
  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