diff --git a/backend/open_webui/routers/retrieval.py b/backend/open_webui/routers/retrieval.py index e3c0bd5f4..f31abd9ff 100644 --- a/backend/open_webui/routers/retrieval.py +++ b/backend/open_webui/routers/retrieval.py @@ -150,8 +150,8 @@ def get_rf( device=DEVICE_TYPE, trust_remote_code=RAG_RERANKING_MODEL_TRUST_REMOTE_CODE, ) - except: - log.error("CrossEncoder error") + except Exception as e: + log.error(f"CrossEncoder: {e}") raise Exception(ERROR_MESSAGES.DEFAULT("CrossEncoder error")) return rf diff --git a/backend/requirements.txt b/backend/requirements.txt index dd7c85932..499eae36d 100644 --- a/backend/requirements.txt +++ b/backend/requirements.txt @@ -54,6 +54,7 @@ elasticsearch==8.17.1 transformers sentence-transformers==3.3.1 +accelerate colbert-ai==0.2.21 einops==0.8.1 diff --git a/pyproject.toml b/pyproject.toml index 4c420af79..52260e45e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -61,6 +61,7 @@ dependencies = [ "transformers", "sentence-transformers==3.3.1", + "accelerate", "colbert-ai==0.2.21", "einops==0.8.1",