revise paddlenlp's version (#2767)

pull/2771/head
TianYuan 2 years ago committed by GitHub
parent 9dd1602c7c
commit 6725bcd823
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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)

@ -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,

@ -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",

Loading…
Cancel
Save