Update on "Use llm_config instead of args in export_llama functions"

Differential Revision: [D75484927](https://our.internmc.facebook.com/intern/diff/D75484927)

[ghstack-poisoned]

Author: jackzhxng

examples/models/llama/TARGETS

@@ -152,6 +152,7 @@ runtime.python_library(
         "//ai_codesign/gen_ai/fast_hadamard_transform:fast_hadamard_transform",
         "//caffe2:torch",
         "//executorch/examples/models/llama/config:llm_config",
+        "//executorch/examples/models/llama/config:llm_config_utils",
         "//executorch/backends/vulkan/_passes:vulkan_passes",
         "//executorch/exir/passes:init_mutable_pass",
         "//executorch/examples/models:model_base",

examples/models/llama/config/llm_config_utils.py

@@ -121,13 +121,17 @@ def convert_args_to_llm_config(args: argparse.Namespace) -> LlmConfig:
     if hasattr(args, "coreml"):
         llm_config.backend.coreml.enabled = args.coreml
     llm_config.backend.coreml.enable_state = getattr(args, "coreml_enable_state", False)
-    llm_config.backend.coreml.preserve_sdpa = getattr(args, "coreml_preserve_sdpa", False)
+    llm_config.backend.coreml.preserve_sdpa = getattr(
+        args, "coreml_preserve_sdpa", False
+    )
     if hasattr(args, "coreml_quantize") and args.coreml_quantize:
         llm_config.backend.coreml.quantize = CoreMLQuantize(args.coreml_quantize)
     if hasattr(args, "coreml_ios"):
         llm_config.backend.coreml.ios = args.coreml_ios
     if hasattr(args, "coreml_compute_units"):
-        llm_config.backend.coreml.compute_units = CoreMLComputeUnit(args.coreml_compute_units)
+        llm_config.backend.coreml.compute_units = CoreMLComputeUnit(
+            args.coreml_compute_units
+        )
 
     # Vulkan
     if hasattr(args, "vulkan"):