Change how to load config of XLNetLMHeadModel (#14746)

2021-12-13 14:34:26 -03:00
parent 15a9d01519
commit 971e36667a
1 changed files with 1 additions and 1 deletions
--- 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))