fix transformers export_onnx no_convert_qat (#524) (#525)

bfineran · web-flow · commit fcc0c553327b · 2022-01-25T16:28:45.000-05:00
diff --git a/src/sparseml/transformers/utils/export.py b/src/sparseml/transformers/utils/export.py
@@ -240,7 +240,7 @@ def main():
         task=args.task,
         model_path=args.model_path,
         sequence_length=args.sequence_length,
-        convert_qat=args.convert_qat,
+        convert_qat=args.no_convert_qat,  # False if flagged
         finetuning_task=args.finetuning_task,
         onnx_file_name=args.onnx_file_name,
     )

Original file line number	Diff line number	Diff line change
`@@ -240,7 +240,7 @@ def main():`
`240`	`240`	`task=args.task,`
`241`	`241`	`model_path=args.model_path,`
`242`	`242`	`sequence_length=args.sequence_length,`
`243`		`- convert_qat=args.convert_qat,`
	`243`	`+ convert_qat=args.no_convert_qat, # False if flagged`
`244`	`244`	`finetuning_task=args.finetuning_task,`
`245`	`245`	`onnx_file_name=args.onnx_file_name,`
`246`	`246`	`)`