UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-04-23 07:45:18 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

chore: include accelerate dependency

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2025-04-06 17:44:05 -07:00
parent a8bc0d65a0
commit 914eb49767
3 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend/open_webui/routers/retrieval.py
View File

@ -150,8 +150,8 @@ def get_rf(
device=DEVICE_TYPE, device=DEVICE_TYPE,
trust_remote_code=RAG_RERANKING_MODEL_TRUST_REMOTE_CODE, trust_remote_code=RAG_RERANKING_MODEL_TRUST_REMOTE_CODE,
) )
except: except Exception as e:
log.error("CrossEncoder error") log.error(f"CrossEncoder: {e}")
raise Exception(ERROR_MESSAGES.DEFAULT("CrossEncoder error")) raise Exception(ERROR_MESSAGES.DEFAULT("CrossEncoder error"))
return rf return rf

1
backend/requirements.txt
View File

@ -54,6 +54,7 @@ elasticsearch==8.17.1
transformers transformers
sentence-transformers==3.3.1 sentence-transformers==3.3.1
accelerate
colbert-ai==0.2.21 colbert-ai==0.2.21
einops==0.8.1 einops==0.8.1

1
pyproject.toml
View File

@ -61,6 +61,7 @@ dependencies = [
"transformers", "transformers",
"sentence-transformers==3.3.1", "sentence-transformers==3.3.1",
"accelerate",
"colbert-ai==0.2.21", "colbert-ai==0.2.21",
"einops==0.8.1", "einops==0.8.1",