fix rerank mode is none (#7496)

This commit is contained in:
Jyong
2024-08-21 16:42:28 +08:00
committed by GitHub
parent a02118d5bc
commit 715eb8fa32

View File

@@ -93,7 +93,7 @@ class DatasetConfigManager:
reranking_model=dataset_configs.get('reranking_model'), reranking_model=dataset_configs.get('reranking_model'),
weights=dataset_configs.get('weights'), weights=dataset_configs.get('weights'),
reranking_enabled=dataset_configs.get('reranking_enabled', True), reranking_enabled=dataset_configs.get('reranking_enabled', True),
rerank_mode=dataset_configs["reranking_mode"], rerank_mode=dataset_configs.get('rerank_mode', 'reranking_model'),
) )
) )