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