fix: fixed typo on loading reranking_mode (#7887)

This commit is contained in:
Fei He 2024-09-02 16:18:47 +08:00 committed by GitHub
parent 7193e189f3
commit a8b9e01b3e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

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