diff --git a/hanlp/components/sts/transformer_sts.py b/hanlp/components/sts/transformer_sts.py
index <HASH>..<HASH> 100644
--- a/hanlp/components/sts/transformer_sts.py
+++ b/hanlp/components/sts/transformer_sts.py
@@ -161,7 +161,10 @@ class TransformerSemanticTextualSimilarity(TorchComponent):
     # noinspection PyMethodOverriding
     def build_model(self, transformer, training=True, **kwargs) -> torch.nn.Module:
         config = AutoConfig_.from_pretrained(transformer, num_labels=1)
-        model = AutoModelForSequenceClassification.from_pretrained(transformer, config=config)
+        if training:
+            model = AutoModelForSequenceClassification.from_pretrained(transformer, config=config)
+        else:
+            model = AutoModelForSequenceClassification.from_config(config)
         return model
 
     def predict(self, data: Union[List[str], List[List[str]]], batch_size: int = None, **kwargs) -> Union[
diff --git a/hanlp/version.py b/hanlp/version.py
index <HASH>..<HASH> 100644
--- a/hanlp/version.py
+++ b/hanlp/version.py
@@ -2,5 +2,5 @@
 # Author: hankcs
 # Date: 2019-12-28 19:26
 
-__version__ = '2.1.0-alpha.45'
+__version__ = '2.1.0-alpha.46'
 """HanLP version"""
Avoid re-downloading Electra model