update codes (#5072)

* update codes

* update codes

* update codes

PaddleNLP/examples/text_classification/pretrained_models/predict.py

@@ -175,8 +175,8 @@ if __name__ == "__main__":
     args.model_type = args.model_type.lower()
     model_class, tokenizer_class = MODEL_CLASSES[args.model_type]
 
-    if args.model_name_or_path == 'ernie_tiny':
-        # ErnieTinyTokenizer is special for ernie_tiny pretained model.
+    if args.model_name_or_path == 'ernie-tiny':
+        # ErnieTinyTokenizer is special for ernie-tiny pretained model.
         tokenizer = ppnlp.transformers.ErnieTinyTokenizer.from_pretrained(
             args.model_name_or_path)
     else:

PaddleNLP/examples/text_matching/sentence_transformers/predict.py

@@ -29,9 +29,8 @@ from model import SentenceTransformer
 MODEL_CLASSES = {
     "bert": (ppnlp.transformers.BertModel, ppnlp.transformers.BertTokenizer),
     'ernie': (ppnlp.transformers.ErnieModel, ppnlp.transformers.ErnieTokenizer),
-    'roberta': (ppnlp.transformers.RobertaModel,
-                ppnlp.transformers.RobertaTokenizer),
-    'electra': (ppnlp.transformers.Electra, ppnlp.transformers.ElectraTokenizer)
+    'roberta':
+    (ppnlp.transformers.RobertaModel, ppnlp.transformers.RobertaTokenizer)
 }
 
 

PaddleNLP/examples/text_matching/sentence_transformers/train.py

@@ -30,8 +30,8 @@ from model import SentenceTransformer
 MODEL_CLASSES = {
     "bert": (ppnlp.transformers.BertModel, ppnlp.transformers.BertTokenizer),
     'ernie': (ppnlp.transformers.ErnieModel, ppnlp.transformers.ErnieTokenizer),
-    'roberta': (ppnlp.transformers.RobertaModel,
-                ppnlp.transformers.RobertaTokenizer),
+    'roberta':
+    (ppnlp.transformers.RobertaModel, ppnlp.transformers.RobertaTokenizer),
 }