From 6725bcd823cf8164520b0495fa88e207b7842b27 Mon Sep 17 00:00:00 2001 From: TianYuan Date: Tue, 27 Dec 2022 15:21:16 +0800 Subject: [PATCH] revise paddlenlp's version (#2767) --- paddlespeech/text/models/ernie_crf/model.py | 2 +- paddlespeech/text/models/ernie_linear/ernie_linear.py | 4 ++-- setup.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) 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",