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/tests/test_export_llama_lib.py

@@ -7,6 +7,7 @@
 import unittest
 
 from executorch.devtools.backend_debug import get_delegation_info
+from executorch.examples.models.llama.config.llm_config import LlmConfig
 from executorch.examples.models.llama.export_llama_lib import (
     _export_llama,
     build_args_parser,
@@ -40,7 +41,8 @@ def test_has_expected_ops_and_op_counts(self):
         args.use_kv_cache = True
         args.verbose = True
 
-        builder = _export_llama(args)
+        llm_config = LlmConfig.from_args(args)
+        builder = _export_llama(llm_config)
         graph_module = builder.edge_manager.exported_program().graph_module
         delegation_info = get_delegation_info(graph_module)