kvaishnavi commited on
Commit
672d733
·
verified ·
1 Parent(s): d7dbe34

Update onnx/builder.py

Browse files
Files changed (1) hide show
  1. 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.matmul_4bits_quantizer",
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),