chore: include accelerate dependency

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

View File

@ -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

View File

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

View File

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