Skip to content

Commit 2e9dd0a

Browse files
committed
Fix formatting
1 parent d004f3c commit 2e9dd0a

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

src/sagemaker/serve/utils/hardware_detector.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -124,14 +124,17 @@ def _total_inference_model_size_mib(model: str, dtype: str) -> int:
124124
output = None
125125
try:
126126
from accelerate.commands.estimate import estimate_command_parser, gather_data
127+
127128
args = estimate_command_parser().parse_args([model, "--dtypes", dtype])
128129

129130
output = gather_data(
130131
args
131132
) # "dtype", "Largest Layer", "Total Size Bytes", "Training using Adam"
132133
except ImportError:
133-
logger.error("To enable Model size calculations: Install HuggingFace extras dependencies "
134-
"using pip install 'sagemaker[huggingface]>=2.212.0'")
134+
logger.error(
135+
"To enable Model size calculations: Install HuggingFace extras dependencies "
136+
"using pip install 'sagemaker[huggingface]>=2.212.0'"
137+
)
135138

136139
if output is None:
137140
raise ValueError(f"Could not get Model size for {model}")

0 commit comments

Comments
 (0)