From 344595bde36dbb4010411e421030091eb1da8a30 Mon Sep 17 00:00:00 2001 From: Lucas Wilkinson Date: Mon, 16 Sep 2024 16:14:07 +0000 Subject: [PATCH] format --- vllm/profiler/visualize_layerwise_profile.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vllm/profiler/visualize_layerwise_profile.py b/vllm/profiler/visualize_layerwise_profile.py index ea27152670d36..880d3f760d6b2 100644 --- a/vllm/profiler/visualize_layerwise_profile.py +++ b/vllm/profiler/visualize_layerwise_profile.py @@ -1,8 +1,8 @@ -import os import argparse import copy import json import math +import os from pathlib import Path from typing import Any, List, Optional, Tuple @@ -427,7 +427,7 @@ def make_plot_title_suffix(profile_json: dict) -> str: output_directory = args.output_directory if args.output_directory else Path( args.json_trace).parent - + if not os.path.exists(output_directory): os.makedirs(output_directory)