mirror of
https://github.com/open-webui/open-webui
synced 2025-04-08 06:35:04 +00:00
chore: include accelerate
dependency
This commit is contained in:
parent
a8bc0d65a0
commit
914eb49767
@ -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
|
||||
|
||||
|
@ -54,6 +54,7 @@ elasticsearch==8.17.1
|
||||
|
||||
transformers
|
||||
sentence-transformers==3.3.1
|
||||
accelerate
|
||||
colbert-ai==0.2.21
|
||||
einops==0.8.1
|
||||
|
||||
|
@ -61,6 +61,7 @@ dependencies = [
|
||||
|
||||
"transformers",
|
||||
"sentence-transformers==3.3.1",
|
||||
"accelerate",
|
||||
"colbert-ai==0.2.21",
|
||||
"einops==0.8.1",
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user