Skip to content

Commit 0dba6be

Browse files
committed
Fix formatting
1 parent 8e89266 commit 0dba6be

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

src/sagemaker/serve/builder/model_builder.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -71,8 +71,10 @@
7171

7272
MIB_CONVERSION_FACTOR = 0.00000095367431640625
7373
MEMORY_BUFFER_MULTIPLIER = 1.2 # 20% buffer
74-
VERSION_DETECTION_ERROR = "Please install accelerate and transformers for HuggingFace (HF) model " \
75-
"size calculations pip install 'sagemaker[huggingface]'"
74+
VERSION_DETECTION_ERROR = (
75+
"Please install accelerate and transformers for HuggingFace (HF) model "
76+
"size calculations pip install 'sagemaker[huggingface]'"
77+
)
7678

7779

7880
# pylint: disable=attribute-defined-outside-init
@@ -736,7 +738,7 @@ def _total_inference_model_size_mib(self):
736738
args = parser.parse_args([self.model, "--dtypes", dtypes])
737739

738740
output = accelerate.commands.estimate.gather_data.gather_data(
739-
args
741+
args
740742
) # "dtype", "Largest Layer", "Total Size Bytes", "Training using Adam"
741743
except ImportError as e:
742744
logger.warning(VERSION_DETECTION_ERROR)

0 commit comments

Comments
 (0)