[BUG]: gemini plugin checkpoint gives empty checkpoint

[FrankLeeeee]

🐛 Describe the bug

When saving the BERT model with gemini checkpoint IO, the checkpoint 00001 is saved but contains no parameters and also absent from the index.json file.

Environment

No response