File size: 198 Bytes
86e4ec2 |
1 2 3 4 5 6 7 8 9 |
import sys
from transformers import T5EncoderModel
in_path = sys.argv[1]
out_path = sys.argv[2]
model = T5EncoderModel.from_pretrained(in_path)
model.half()
model.save_pretrained(out_path) |