Skip to content

Commit 3862947

Browse files
authored
Revert "Support more model outputs for BERT/ERNIE/RoBERTa (#2583)" (#2647)
This reverts commit dab5e26.
1 parent dab5e26 commit 3862947

File tree

5 files changed

+182
-1092
lines changed

5 files changed

+182
-1092
lines changed

paddlenlp/transformers/__init__.py

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -120,11 +120,4 @@
120120
if is_faster_tokenizer_available():
121121
from .bert.faster_tokenizer import *
122122
from .ernie.faster_tokenizer import *
123-
from .tinybert.faster_tokenizer import *
124-
125-
# patches of paddle.nn.Transformer to get all hidden_states and attentions
126-
import paddle
127-
from .model_outputs import _transformer_encoder_layer_fwd, _transformer_encoder_fwd
128-
129-
paddle.nn.TransformerEncoderLayer.forward = _transformer_encoder_layer_fwd
130-
paddle.nn.TransformerEncoder.forward = _transformer_encoder_fwd
123+
from .tinybert.faster_tokenizer import *

0 commit comments

Comments
 (0)