fix load_weight_prefix (#15101)

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2022-02-03 16:11:53 +01:00
parent 71dccd0774
commit f5d98da29e
2 changed files with 2 additions and 2 deletions
--- 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")