From f5d98da29ec7f3c2701356f1cd7d4e58fae466bd Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Thu, 3 Feb 2022 16:11:53 +0100 Subject: [PATCH] fix load_weight_prefix (#15101) Co-authored-by: ydshieh --- .../models/encoder_decoder/modeling_tf_encoder_decoder.py | 2 +- tests/test_modeling_tf_encoder_decoder.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/encoder_decoder/modeling_tf_encoder_decoder.py b/src/transformers/models/encoder_decoder/modeling_tf_encoder_decoder.py index fb5c2ab725..63ce2c87ac 100644 --- a/src/transformers/models/encoder_decoder/modeling_tf_encoder_decoder.py +++ b/src/transformers/models/encoder_decoder/modeling_tf_encoder_decoder.py @@ -155,7 +155,7 @@ class TFEncoderDecoderModel(TFPreTrainedModel): """ config_class = EncoderDecoderConfig base_model_prefix = "encoder_decoder" - load_weight_prefix = "tf_encoder_decoder_model_1" + load_weight_prefix = "tf_encoder_decoder_model" def __init__( self, diff --git a/tests/test_modeling_tf_encoder_decoder.py b/tests/test_modeling_tf_encoder_decoder.py index 96f2b81554..d5b0a9dd61 100644 --- a/tests/test_modeling_tf_encoder_decoder.py +++ b/tests/test_modeling_tf_encoder_decoder.py @@ -958,7 +958,7 @@ class TFEncoderDecoderModelSaveLoadTests(unittest.TestCase): @slow def test_encoder_decoder_from_pretrained(self): - load_weight_prefix = "tf_encoder_decoder_model_1" + load_weight_prefix = TFEncoderDecoderModel.load_weight_prefix config = self.get_encoder_decoder_config() encoder_tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")