Skip to content

Commit

Permalink
format
Browse files Browse the repository at this point in the history
  • Loading branch information
LucasWilkinson committed Sep 16, 2024
1 parent 0238110 commit d857de9
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions vllm/profiler/visualize_layerwise_profile.py
Original file line number Diff line number Diff line change
@@ -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

Expand Down Expand Up @@ -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)

Expand Down

0 comments on commit d857de9

Please sign in to comment.