Timothy J. Baek 7 месяцев назад
Родитель
Сommit
8206c47a47
1 измененных файлов с 2 добавлено и 1 удалено
  1. 2 1
      backend/open_webui/apps/retrieval/main.py

+ 2 - 1
backend/open_webui/apps/retrieval/main.py

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