Update onnx/builder.py
Browse files- onnx/builder.py +1 -1
onnx/builder.py
CHANGED
@@ -95,7 +95,7 @@ def build_vision(args):
|
|
95 |
# ORT 4-bits quantizer
|
96 |
fpath_4 = os.path.join(args.output, filename)
|
97 |
cmd = [
|
98 |
-
f"{sys.executable}", "-m", "onnxruntime.quantization.
|
99 |
"--input_model", fpath_3,
|
100 |
"--output_model", fpath_4,
|
101 |
"--block_size", str(32),
|
|
|
95 |
# ORT 4-bits quantizer
|
96 |
fpath_4 = os.path.join(args.output, filename)
|
97 |
cmd = [
|
98 |
+
f"{sys.executable}", "-m", "onnxruntime.quantization.matmul_nbits_quantizer",
|
99 |
"--input_model", fpath_3,
|
100 |
"--output_model", fpath_4,
|
101 |
"--block_size", str(32),
|