[run_language_modeling] Fix: initialize a new model from a config object
This commit is contained in:
1 parent
f8823bad9a
commit
eaabaaf750
1 file changed
+1
-1
@@ -710,7 +710,7 @@ def main():
|
||||
)
|
||||
else:
|
||||
logger.info("Training new model from scratch")
|
||||
model = AutoModelWithLMHead(config=config)
|
||||
model = AutoModelWithLMHead.from_config(config)
|
||||
|
||||
model.to(args.device)
|
||||
|
||||
|
||||
Reference in new issue
Block a user