9 lines
198 B
Python
9 lines
198 B
Python
|
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)
|