From f761140a97e54247fd56c93c2e0732da49a10d08 Mon Sep 17 00:00:00 2001 From: "zhangzhicheng.zzc" Date: Sat, 11 Feb 2023 01:04:47 +0000 Subject: [PATCH] [to #42322933] without setting in config, textranking's num_labels default as 1 Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11629915 * without setting, textranking's num_labels default as 1 --- modelscope/models/nlp/task_models/text_ranking.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/modelscope/models/nlp/task_models/text_ranking.py b/modelscope/models/nlp/task_models/text_ranking.py index c43453bf..e883d76c 100644 --- a/modelscope/models/nlp/task_models/text_ranking.py +++ b/modelscope/models/nlp/task_models/text_ranking.py @@ -35,7 +35,9 @@ class ModelForTextRanking(EncoderModel): label2id = parse_label_mapping(model_dir) if label2id is not None and len(label2id) > 0: num_labels = len(label2id) - self.id2label = {id: label for label, id in label2id.items()} + self.id2label = {id: label for label, id in label2id.items()} + elif label2id is None: + num_labels = 1 kwargs['num_labels'] = num_labels super().__init__(model_dir, *args, **kwargs)