changes (#5669)

PaddlePaddle · Apr 14, 2023 · 52aa271 · 52aa271
1 parent 848f5c1
commit 52aa271
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddlenlp/transformers/gpt/modeling.py b/paddlenlp/transformers/gpt/modeling.py
@@ -878,7 +878,7 @@ def forward(
             encoder_outputs, cached_kvs = outputs[:2]
         else:
             encoder_outputs = outputs
-        logits = self.decoder(encoder_outputs)
+        logits = self.lm_head(encoder_outputs)
 
         if use_cache:
             return logits, cached_kvs