chore: include accelerate dependency
This commit is contained in:
1 parent
a8bc0d65a0
commit
914eb49767
3 files changed
+4
-2
No files matched your search
@@ -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
|
||||
|
||||
|
||||
Reference in new issue
Block a user