Change how to load config of XLNetLMHeadModel (huggingface#14746)

josutk · web-flow · commit 971e36667a79 · 2021-12-13T12:34:26.000-05:00
diff --git 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))
 

Original file line number	Diff line number	Diff line change
`@@ -348,7 +348,7 @@ def main():`
`348`	`348`	`)`
`349`	`349`	`else:`
`350`	`350`	`logger.info("Training new model from scratch")`
`351`		`- model = XLNetLMHeadModel.from_config(config)`
	`351`	`+ model = XLNetLMHeadModel(config)`
`352`	`352`
`353`	`353`	`model.resize_token_embeddings(len(tokenizer))`
`354`	`354`