gtr-t5-base / convert_to_fp16.py
nreimers's picture
add convert files
967b085
raw history blame
No virus
198 Bytes
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)