Commit eaabaaf7 authored by Julien Chaumond's avatar Julien Chaumond
Browse files

[run_language_modeling] Fix: initialize a new model from a config object

parent f8823bad
...@@ -710,7 +710,7 @@ def main(): ...@@ -710,7 +710,7 @@ def main():
) )
else: else:
logger.info("Training new model from scratch") logger.info("Training new model from scratch")
model = AutoModelWithLMHead(config=config) model = AutoModelWithLMHead.from_config(config)
model.to(args.device) model.to(args.device)
......
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