Model Memory Calculator not working for gated models

It seems the issue is that the version of Transformers in the Model Memory Calculator space is outdated, so it can’t load the new model architecture. My test space: https://huggingface.co/spaces/John6666/model-memory-usage-mod