"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "773849415aff693d528a28617441284c0e541a6f"
Unverified Commit f5d98da2 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

fix load_weight_prefix (#15101)


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 71dccd07
......@@ -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,
......
......@@ -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")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment