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),
|