Gatortron-OG or S models. Trouble loading into pytorch

For the two new Gatortron models I see issues loading the checkpoint file in pytorch. The call torch.load(‘dir/’, map_location=‘cpu’) should work but I get this error below.

[enforce fail at] . PytorchStreamReader failed reading zip archive: failed finding central directory.


Hi rich11,

Thank you for your interest in the Gatortron-OG / Gatortron-S pre-trained language models.

Seems the models on NGC were corrupted - please delete your original download and re-download the latest available model.


Thank you. I will take a look.