bpt-30b / convert.py
rob
v0.3
d2bef5f
raw
history blame
127 Bytes
from transformers import AutoModel
model = AutoModel.from_pretrained("./")
model.save_pretrained('./', max_shard_size="5GB")