chore: include accelerate dependency

This commit is contained in:
Timothy Jaeryang Baek committed 2025-04-06 17:44:05 -07:00
1 parent a8bc0d65a0
commit 914eb49767
3 files changed
+4 -2

No files matched your search

+2 -2
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