import onnx, time, sys
from onnxruntime.transformers.onnx_model import OnnxModel
D = "/Users/shreyas/work/rnd/gliner2/work/export/onnx"
src, dst, block = sys.argv[1], sys.argv[2], sys.argv[3:]
t0 = time.time()
m = OnnxModel(onnx.load(D + "/" + src))
m.convert_float_to_float16(keep_io_types=True, op_block_list=block or None, use_symbolic_shape_infer=False)
m.save_model_to_file(D + "/" + dst, use_external_data_format=True, all_tensors_to_one_file=True)
print(dst, "done", round(time.time()-t0,1), "s")