From 30b6c8854d1af4aa5cebfa44570453bc7980b00c Mon Sep 17 00:00:00 2001 From: Samrudhi Sharma Date: Thu, 14 Mar 2024 19:59:21 +0000 Subject: [PATCH] Fix formatting --- src/sagemaker/serve/builder/model_builder.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/src/sagemaker/serve/builder/model_builder.py b/src/sagemaker/serve/builder/model_builder.py index 678db430bc..15c0279b63 100644 --- 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)