Fix formatting

aws · Mar 14, 2024 · 30b6c88 · 30b6c88
1 parent 3b5475e
commit 30b6c88
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/sagemaker/serve/builder/model_builder.py b/src/sagemaker/serve/builder/model_builder.py
@@ -71,8 +71,10 @@
 
 MIB_CONVERSION_FACTOR = 0.00000095367431640625
 MEMORY_BUFFER_MULTIPLIER = 1.2  # 20% buffer
-VERSION_DETECTION_ERROR = "Please install accelerate and transformers for HuggingFace (HF) model " \
-                          "size calculations pip install 'sagemaker[huggingface]'"
+VERSION_DETECTION_ERROR = (
+    "Please install accelerate and transformers for HuggingFace (HF) model "
+    "size calculations pip install 'sagemaker[huggingface]'"
+)
 
 
 # pylint: disable=attribute-defined-outside-init
@@ -736,7 +738,7 @@ def _total_inference_model_size_mib(self):
             args = parser.parse_args([self.model, "--dtypes", dtypes])
 
             output = accelerate.commands.estimate.gather_data.gather_data(
-                    args
+                args
             )  # "dtype", "Largest Layer", "Total Size Bytes", "Training using Adam"
         except ImportError as e:
             logger.warning(VERSION_DETECTION_ERROR)