diff --git a/modelscope/models/nlp/sbert_for_sentence_similarity.py b/modelscope/models/nlp/sbert_for_sentence_similarity.py
index db469f4f..e893a301 100644
--- a/modelscope/models/nlp/sbert_for_sentence_similarity.py
+++ b/modelscope/models/nlp/sbert_for_sentence_similarity.py
@@ -18,6 +18,6 @@ class SbertForSentenceSimilarity(SbertForSequenceClassificationBase):
             model_cls (Optional[Any], optional): model loader, if None, use the
                 default loader to load model weights, by default None.
         """
-        super().__init__(model_dir, *args, **kwargs)
+        super().__init__(model_dir, *args, model_args={"num_labels": 2}, **kwargs)
         self.model_dir = model_dir
         assert self.model.config.num_labels == 2
diff --git a/modelscope/models/nlp/sbert_for_sentiment_classification.py b/modelscope/models/nlp/sbert_for_sentiment_classification.py
index 4683b079..10bfaa0f 100644
--- a/modelscope/models/nlp/sbert_for_sentiment_classification.py
+++ b/modelscope/models/nlp/sbert_for_sentiment_classification.py
@@ -19,5 +19,5 @@ class SbertForSentimentClassification(SbertForSequenceClassificationBase):
             model_cls (Optional[Any], optional): model loader, if None, use the
                 default loader to load model weights, by default None.
         """
-        super().__init__(model_dir, *args, **kwargs)
+        super().__init__(model_dir, *args, model_args={"num_labels": 2}, **kwargs)
         assert self.model.config.num_labels == 2