File tree Expand file tree Collapse file tree 1 file changed +5
-3
lines changed
src/sagemaker/serve/builder Expand file tree Collapse file tree 1 file changed +5
-3
lines changed Original file line number Diff line number Diff line change 73
73
MEMORY_BUFFER_MULTIPLIER = 1.2 # 20% buffer
74
74
VERSION_DETECTION_ERROR = (
75
75
"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]'"
77
77
)
78
78
79
79
@@ -734,10 +734,12 @@ def _total_inference_model_size_mib(self):
734
734
import accelerate .commands .estimate .gather_data
735
735
736
736
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
738
740
args = parser .parse_args ([self .model , "--dtypes" , dtypes ])
739
741
740
- output = accelerate .commands .estimate .gather_data .gather_data (
742
+ output = accelerate .commands .estimate .gather_data .gather_data ( # pylint: disable=E1101
741
743
args
742
744
) # "dtype", "Largest Layer", "Total Size Bytes", "Training using Adam"
743
745
except ImportError as e :
You can’t perform that action at this time.
0 commit comments