Skip to content

Commit

Permalink
Update src/python/models/export.py
Browse files Browse the repository at this point in the history
Co-authored-by: kunal-vaishnavi <[email protected]>
  • Loading branch information
yufenglee and kunal-vaishnavi authored Feb 15, 2024
1 parent 68bf943 commit e10d38f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/python/models/export.py
Original file line number Diff line number Diff line change
Expand Up @@ -104,7 +104,7 @@ def __init__(self, config, io_dtype, onnx_dtype, ep, cache_dir, extra_options):
self.quant_attrs = {
"int4": {
"block_size": int(extra_options["int4_block_size"]) if "int4_block_size" in extra_options else 32,
"accuracy_level": int(extra_options["int4_accuracy_level"]) if "int_accuracy_level" in extra_options else None,
"accuracy_level": int(extra_options["int4_accuracy_level"]) if "int4_accuracy_level" in extra_options else None,
}
}

Expand Down

0 comments on commit e10d38f

Please sign in to comment.