Enhance customize.py to handle dynamic exports

Added dynamic configuration handling for llama2-70b model.

Author: anandhu-eng

script/add-custom-nvidia-system/customize.py

@@ -2,7 +2,8 @@
 from utils import *
 import os
 import shutil
-
+import importlib.util
+import json
 
 def preprocess(i):
 
@@ -43,6 +44,26 @@ def postprocess(i):
         src_file = os.path.join(tmp_script_path, "dummy_config.py")
 
         dest_file = os.path.join(target_dir, f"{mlc_model}.py")
-        shutil.copy(src_file, dest_file)
+
+        dummy_config_path = os.path.join(tmp_script_path, "dummy_config.py")
+        spec = importlib.util.spec_from_file_location("dummy_config", dummy_config_path)
+        dummy_module = importlib.util.module_from_spec(spec)
+        spec.loader.exec_module(dummy_module)
+
+        EXPORTS = dummy_module.EXPORTS
+
+        # --- Example dynamic insertion ---
+        if mlc_model == "llama2-70b":
+            for k, v in EXPORTS.items():
+                if isinstance(v, dict):
+                    v.setdefault('llm_fields.llm_gen_config_path',
+                                 'code/llama2-70b/tensorrt/generation_config.json')
+
+        # --- Write modified config to destination ---
+        with open(dest_file, "w") as f:
+            f.write("# Auto-generated config\n\n")
+            f.write("EXPORTS = ")
+            f.write(repr(EXPORTS))
+            f.write("\n")
 
     return {'return': 0}