From 914eb49767f6c65eb1e209d8883c5bd308ba63a4 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Sun, 6 Apr 2025 17:44:05 -0700 Subject: [PATCH] chore: include `accelerate` dependency --- backend/open_webui/routers/retrieval.py | 4 ++-- backend/requirements.txt | 1 + pyproject.toml | 1 + 3 files changed, 4 insertions(+), 2 deletions(-) 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",