diff --git a/tests/integration/defs/perf/pytorch_model_config.py b/tests/integration/defs/perf/pytorch_model_config.py
index 476e23079c7..ba01027d465 100644
--- a/tests/integration/defs/perf/pytorch_model_config.py
+++ b/tests/integration/defs/perf/pytorch_model_config.py
@@ -150,7 +150,11 @@ def get_model_yaml_config(model_label: str,
                 'trtllm_modules_to_hf_modules': {
                     "attn_q": "q_proj",
                     "attn_k": "k_proj",
-                    "attn_v": "v_proj"
+                    "attn_v": "v_proj",
+                    "attn_dense": "o_proj",
+                    "mlp_h_to_4h": "gate_proj",
+                    "mlp_4h_to_h": "down_proj",
+                    "mlp_gate": "up_proj",
                 }
             }
         }