Gatortron-OG or S models. Trouble loading into pytorch

Healthcare Clara Train: AI-Assisted Annotation SDK
#1

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

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

-RIchard

#2

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.

Thanks

#3

Thank you. I will take a look.