Improve readability and improve make less assumptions about checkpoint format
Improve readability and improve make less assumptions about checkpoint format

huggingface/transformers,huggingface/transformers,huggingface/transformers,huggingface/pytorch-transformers,huggingface/transformers