Skip to content

Commit 0629384

Browse files
committed
Minor update to description
1 parent 30b6c88 commit 0629384

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
@@ -73,7 +73,7 @@
7373
MEMORY_BUFFER_MULTIPLIER = 1.2 # 20% buffer
7474
VERSION_DETECTION_ERROR = (
7575
"Please install accelerate and transformers for HuggingFace (HF) model "
76-
"size calculations pip install 'sagemaker[huggingface]'"
76+
"size calculations e.g. pip install 'sagemaker[huggingface]'"
7777
)
7878

7979

@@ -734,10 +734,12 @@ def _total_inference_model_size_mib(self):
734734
import accelerate.commands.estimate.gather_data
735735

736736
dtypes = self.env_vars.get("dtypes", "float32")
737-
parser = accelerate.commands.estimate.estimate_command_parser.estimate_command_parser()
737+
parser = (
738+
accelerate.commands.estimate.estimate_command_parser.estimate_command_parser()
739+
) # pylint: disable=E1101
738740
args = parser.parse_args([self.model, "--dtypes", dtypes])
739741

740-
output = accelerate.commands.estimate.gather_data.gather_data(
742+
output = accelerate.commands.estimate.gather_data.gather_data( # pylint: disable=E1101
741743
args
742744
) # "dtype", "Largest Layer", "Total Size Bytes", "Training using Adam"
743745
except ImportError as e:

0 commit comments

Comments
 (0)