diff --git a/paddlespeech/text/models/ernie_crf/model.py b/paddlespeech/text/models/ernie_crf/model.py
index d1ce8099..19f591b8 100644
--- a/paddlespeech/text/models/ernie_crf/model.py
+++ b/paddlespeech/text/models/ernie_crf/model.py
@@ -27,7 +27,7 @@ class ErnieCrf(nn.Layer):
                  **kwargs):
         super().__init__()
         self.ernie = ErnieForTokenClassification.from_pretrained(
-            pretrained_token, num_classes=num_classes, **kwargs)
+            pretrained_token, num_labels=num_classes, **kwargs)
         self.num_classes = num_classes
         self.crf = LinearChainCrf(
             self.num_classes, crf_lr=crf_lr, with_start_stop_tag=False)
diff --git a/paddlespeech/text/models/ernie_linear/ernie_linear.py b/paddlespeech/text/models/ernie_linear/ernie_linear.py
index c450a904..48a261ab 100644
--- a/paddlespeech/text/models/ernie_linear/ernie_linear.py
+++ b/paddlespeech/text/models/ernie_linear/ernie_linear.py
@@ -43,9 +43,9 @@ class ErnieLinear(nn.Layer):
                 num_classes, int
             ) and num_classes > 0, 'Argument `num_classes` must be an integer.'
             self.ernie = ErnieForTokenClassification.from_pretrained(
-                pretrained_token, num_classes=num_classes, **kwargs)
+                pretrained_token, num_labels=num_classes, **kwargs)
 
-        self.num_classes = self.ernie.num_classes
+        self.num_classes = self.ernie.num_labels
         self.softmax = nn.Softmax()
 
     def forward(self,
diff --git a/setup.py b/setup.py
index 3b58adf3..e55f8454 100644
--- a/setup.py
+++ b/setup.py
@@ -47,7 +47,7 @@ base = [
     "onnxruntime==1.11.0",
     "opencc",
     "pandas",
-    "paddlenlp>=2.4.3, <=2.4.5",
+    "paddlenlp>=2.4.8",
     "paddlespeech_feat",
     "Pillow>=9.0.0",
     "praatio==5.0.0",