From 052da3d7d0ae844bddf4405537873215dc4a4243 Mon Sep 17 00:00:00 2001
From: Zhong Hui <zhonghui.net@gmail.com>
Date: Fri, 5 Feb 2021 17:10:48 +0800
Subject: [PATCH] Quick fix typo (#5273)

---
 PaddleNLP/paddlenlp/transformers/ernie/modeling.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/PaddleNLP/paddlenlp/transformers/ernie/modeling.py b/PaddleNLP/paddlenlp/transformers/ernie/modeling.py
index 2b2cfe7f..fe032e38 100644
--- a/PaddleNLP/paddlenlp/transformers/ernie/modeling.py
+++ b/PaddleNLP/paddlenlp/transformers/ernie/modeling.py
@@ -327,7 +327,7 @@ class ErnieForTokenClassification(ErniePretrainedModel):
 class ErnieForPretraining(ErniePretrainedModel):
     def __init__(self, ernie):
         super(ErnieForPretraining, self).__init__()
-        self.ernie = bert
+        self.ernie = ernie
         self.cls = ErniePretrainingHeads(
             self.ernie.config["hidden_size"],
             self.ernie.config["vocab_size"],
-- 
GitLab