Fix export script to use pt2e if not selected

Summary:
Fix export script to use pt2e if not selected
- This is useful for coreml export since it's not using pt2e quantizer

Differential Revision: D68522589

Author: Di Xu (SWE)

examples/models/llama/export_llama_lib.py

@@ -662,17 +662,21 @@ def _validate_args(args):
 
 def _export_llama(args) -> LLMEdgeManager:  # noqa: C901
     _validate_args(args)
-    pt2e_quant_params, quantizers, quant_dtype = get_quantizer_and_quant_params(args)
 
     # export_to_edge
     builder_exported = _prepare_for_llama_export(args).export()
 
     if args.export_only:
         exit()
 
-    builder_exported_to_edge = builder_exported.pt2e_quantize(
-        quantizers
-    ).export_to_edge()
+    if args.pt2e_quantize:
+        pt2e_quant_params, quantizers, quant_dtype = get_quantizer_and_quant_params(args)
+        builder_exported_to_edge = builder_exported.pt2e_quantize(
+            quantizers
+        ).export_to_edge()
+    else:
+        pt2e_quant_params = None
+        builder_exported_to_edge = builder_exported.export_to_edge()
 
     modelname = builder_exported_to_edge.modelname