This commit is contained in:
Timothy Jaeryang Baek committed 2025-07-14 14:05:06 +04:00
1 parent 0013f5c1fc
commit 18bd83413b
3 files changed
+16 -4

No files matched your search

+2
View File
@@ -446,6 +446,8 @@ def get_embedding_function(
def get_reranking_function(reranking_engine, reranking_model, reranking_function):
if reranking_function is None:
return None
if reranking_engine == "external":
return lambda sentences, user=None: reranking_function.predict(
sentences, user=user