from transformers import AutoModel; AutoModel.from_pretrained('./', from_flax=True).save_pretrained('./')