From f337af3800200f8c34fb9a3506c9ec839d162cc3 Mon Sep 17 00:00:00 2001 From: krishung5 Date: Mon, 22 Apr 2024 19:08:13 -0700 Subject: [PATCH] Format --- build.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/build.py b/build.py index ac50c9d1f4d..4e012f2011b 100755 --- a/build.py +++ b/build.py @@ -820,7 +820,9 @@ def fastertransformer_cmake_args(): def tensorrtllm_cmake_args(images): cmake_script.cmd("apt-get update && apt-get install -y libcudnn8-dev && ldconfig") - cmake_script.cmd("python3 ../tensorrt_llm/scripts/build_wheel.py --trt_root /usr/local/tensorrt") + cmake_script.cmd( + "python3 ../tensorrt_llm/scripts/build_wheel.py --trt_root /usr/local/tensorrt" + ) cargs = [ cmake_backend_arg( "tensorrtllm",