d2bef5f
1
2
3
4
5
6
from transformers import AutoModel model = AutoModel.from_pretrained("./") model.save_pretrained('./', max_shard_size="5GB")