Skip to content

Commit bf1383e

Browse files
fix linter errors
1 parent 7f03d3e commit bf1383e

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

torchao/float8/benchmarking/float8_training_benchmark.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ LOG_FILE="/tmp/float8_training_log.txt"
1313
# validate user has specified torchtitan root directory
1414
if [ -z "${TORCHTITAN_ROOT}" ]; then
1515
echo "Error: TORCHTITAN environment variable is not set. Please set it before running this script."
16-
echo "Usage: TORCHTITAN_ROOT=<directory> ./llama3_8b_float8_training_benchmark.sh"
16+
echo "Usage: TORCHTITAN_ROOT=<directory> ./float8_training_benchmark.sh"
1717
echo "Optional parameters configurable via environment variables:"
1818
echo " * RECIPE: rowwise|tensorwise. defaults to tensorwise."
1919
echo " * BATCH_SIZE: defaults to 1."

torchao/float8/benchmarking/parse_torchtitan_logs.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
Usage:
77
python parse_torchtitan_logs.py --log-file <log_file_path>
88
"""
9+
910
import os
1011
import re
1112
import statistics

0 commit comments

Comments
 (0)