From 971e36667a79666356d5eb98d2b748a5a0e24a68 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Josu=C3=A9=20Nascimento?= Date: Mon, 13 Dec 2021 14:34:26 -0300 Subject: [PATCH] Change how to load config of XLNetLMHeadModel (#14746) --- examples/pytorch/language-modeling/run_plm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/pytorch/language-modeling/run_plm.py b/examples/pytorch/language-modeling/run_plm.py index 17191ee10a..3c307c9fb8 100755 --- a/examples/pytorch/language-modeling/run_plm.py +++ b/examples/pytorch/language-modeling/run_plm.py @@ -348,7 +348,7 @@ def main(): ) else: logger.info("Training new model from scratch") - model = XLNetLMHeadModel.from_config(config) + model = XLNetLMHeadModel(config) model.resize_token_embeddings(len(tokenizer))