Update onnxruntime/python/tools/transformers/models/llama/convert_to_…

…onnx.py Co-authored-by: kunal-vaishnavi <[email protected]>
microsoft · Aug 15, 2024 · 70f06de · 70f06de
1 parent bc8b872
commit 70f06de
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/onnxruntime/python/tools/transformers/models/llama/convert_to_onnx.py b/onnxruntime/python/tools/transformers/models/llama/convert_to_onnx.py
@@ -1001,6 +1001,8 @@ def main():
                         logger.info(f"The ONNX model at {fp_path} has been quantized to int4 and saved at {int4_path}!")
                         remove_existing_model(fp_path)
         barrier()
+
+    # Don't run parity checker for Dynamo export currently
     if args.use_dynamo_export:
         return