File size: 127 Bytes
d2bef5f
 
 
 
 
1
2
3
4
5
6
from transformers import AutoModel

model = AutoModel.from_pretrained("./")

model.save_pretrained('./', max_shard_size="5GB")