pythonhuggingface-transformerslanguage-modelgpt-2

Scripts missing for GPT-2 fine tune, and inference in Hugging-face GitHub?


I am following the documentation on the hugging face website, in there they say that to fine-tune GPT-2 I should use the script run_lm_finetuning.py for fine-tuning, and the script run_generation.py for inference. However, both scripts don't actually exist on GitHub anymore.

Does anybody know whether the documentation is outdated? or where to find those two scripts?

Thanks


Solution

  • It looks like they've been moved around a couple times and the docs are indeed out of date, the current version can be found in run_language_modeling.py here https://github.com/huggingface/transformers/tree/master/examples/language-modeling