Change how to load config of XLNetLMHeadModel (#14746)

971e3666 · Josué Nascimento · GitHub · 15a9d015 · 971e3666
Unverified Commit 971e3666 authored Dec 13, 2021 by Josué Nascimento Committed by GitHub Dec 13, 2021
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/pytorch/language-modeling/run_plm.py examples/pytorch/language-modeling/run_plm.py +1 -1

No files found.
--- 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))