修复获取节点后处理机制BUG

This commit is contained in:
2024-08-19 09:14:54 +08:00
parent 26ecb256ce
commit 806b694b37
+1 -1
View File
@@ -13,7 +13,7 @@ def get_node_postprocessors():
rerank_model = os.getenv("RERANK_MODEL") rerank_model = os.getenv("RERANK_MODEL")
rerank_url = os.getenv("RERANK_BASE_URL") rerank_url = os.getenv("RERANK_BASE_URL")
postprocess = None postprocess = None
if rerank_model is None: if rerank_model is not None:
postprocess = [XinferenceRerank(rerank_model, rerank_url)] postprocess = [XinferenceRerank(rerank_model, rerank_url)]
return postprocess return postprocess