diff --git a/transformer_nuggets/utils/benchmark.py b/transformer_nuggets/utils/benchmark.py index f2ad33c..32a0574 100644 --- a/transformer_nuggets/utils/benchmark.py +++ b/transformer_nuggets/utils/benchmark.py @@ -281,7 +281,7 @@ def profiler( output_dir = path.parent output_dir.mkdir(parents=True, exist_ok=True) - logger.info(f"💾 Trace file 📄 saved to: {bcolors.OKGREEN}{output_dir}{bcolors.ENDC}") + logger.info(f"💾 Trace file 📄 saved to: {bcolors.OKGREEN}{path}{bcolors.ENDC}") def trace_handler(prof) -> None: prof.export_chrome_trace(path.as_posix())