TypeError: __init__() got an unexpected keyword argument 'cache_dir'

I'm fine tuning with the example `run_language_modeling.py`as follows.
```shell
python run_language_modeling.py --output_dir=output_dir --model_type gpt2 --model_name_or_path distilgpt2  --do_train  --train_data_file=xxx.data.txt
```
It failed with error 
```shell
Traceback (most recent call last):
  File "run_language_modeling.py", line 313, in <module>
    main()
  File "run_language_modeling.py", line 242, in main
    get_dataset(data_args, tokenizer=tokenizer, cache_dir=model_args.cache_dir) if training_args.do_train else None
  File "run_language_modeling.py", line 143, in get_dataset
    cache_dir=cache_dir,
TypeError: __init__() got an unexpected keyword argument 'cache_dir'
```
Could you please tell me how to use it ? Thanks a lot.
<!-- A clear and concise description of what you would expect to happen. -->

Solved with  https://github.com/huggingface/transformers/issues/319