Transformers - How to load pytorch_model.bin in torch version 1.4.0 (Albert)

Question

torch version 1.4.0 I execute run_language_modeling.py and save the model. However, when I load the saved model, "OSError: Unable to load weights from pytorch checkpoin