def get_trtllm_engines()

in src/optimum/nvidia/hub.py [0:0]

• 6 lines of code
• 2 McCabe index (conditional complexity)

def get_trtllm_engines(model_id: str, device: str, dtype: str):
    if (workspace := Workspace.from_hub_cache(model_id, device)).engines_path.exists():
        return workspace.engines_path

    return get_trtllm_artifact(
        model_id, [f"{device}/{dtype}/**/{PATH_FOLDER_ENGINES}/*.engine"]
    )