saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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

Browse source
This commit is contained in:
Julien Chaumond 2020-03-24 17:56:40 -04:00
parent f8823bad9a
commit eaabaaf750
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
examples/run_language_modeling.py
View file

@ -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)