File size: 232 Bytes
5285b72
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
#!/usr/bin/env python

import sys
from pathlib import Path
from transformers.convert_graph_to_onnx import quantize

input_file = sys.argv[1]
quantized_model_path = quantize(Path(input_file))
quantized_model_path.replace(input_file)