kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-13 22:18:36 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

tests: Fix lora perf test (#5875)

Browse Source 
Signed-off-by: Amir Klein <203507526+amirkl94@users.noreply.github.com>
This commit is contained in:
amirkl94 2025-07-10 11:56:46 +03:00 committed by GitHub
parent ff9aabb038
commit cd7aeec061
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/integration/defs/perf/pytorch_model_config.py
View File

@ -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",
}
}
}