optimum_benchmark/backends/tensorrt_llm/utils.py (
1
lines of code) (
raw
):
MODEL_TYPE_TO_TRTLLMMODELS = {"llama": "optimum.nvidia.models.llama.LlamaForCausalLM"}