Skip to content

Commit 36e0d7a

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

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/sagemaker/serve/builder/model_builder.py

Lines changed: 4 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,11 @@ 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 = accelerate.commands.estimate.estimate_command_parser\
738+
.estimate_command_parser() # pylint: disable=E1101
738739
args = parser.parse_args([self.model, "--dtypes", dtypes])
739740

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

0 commit comments

Comments
 (0)