Minor update to description

aws · Mar 14, 2024 · 0629384 · 0629384
1 parent 30b6c88
commit 0629384
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
@@ -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: