From 0629384a880607cf2e9febbc55a3b774657f9a5e Mon Sep 17 00:00:00 2001 From: Samrudhi Sharma Date: Thu, 14 Mar 2024 20:19:47 +0000 Subject: [PATCH] Minor update to description --- 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 15c0279b63a..ef7d6346577 100644 --- a/src/sagemaker/serve/builder/model_builder.py +++ b/src/sagemaker/serve/builder/model_builder.py @@ -73,7 +73,7 @@ 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]'" + "size calculations e.g. pip install 'sagemaker[huggingface]'" ) @@ -734,10 +734,12 @@ def _total_inference_model_size_mib(self): import accelerate.commands.estimate.gather_data dtypes = self.env_vars.get("dtypes", "float32") - parser = accelerate.commands.estimate.estimate_command_parser.estimate_command_parser() + parser = ( + accelerate.commands.estimate.estimate_command_parser.estimate_command_parser() + ) # pylint: disable=E1101 args = parser.parse_args([self.model, "--dtypes", dtypes]) - output = accelerate.commands.estimate.gather_data.gather_data( + output = accelerate.commands.estimate.gather_data.gather_data( # pylint: disable=E1101 args ) # "dtype", "Largest Layer", "Total Size Bytes", "Training using Adam" except ImportError as e: