Skip to content

Commit e6209de

Browse files
author
Bryannah Hernandez
committed
CR Fixes for InferenceSpec MMS
1 parent 895fcf2 commit e6209de

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

src/sagemaker/serve/builder/transformers_builder.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -279,8 +279,6 @@ def _build_transformers_env(self):
279279
if self.mode == Mode.LOCAL_CONTAINER:
280280
self._prepare_for_mode()
281281

282-
logger.info("Model configuration %s", self.pysdk_model)
283-
284282
return self.pysdk_model
285283

286284
def _set_instance(self, **kwargs):

src/sagemaker/serve/model_server/multi_model_server/prepare.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ def prepare_for_mms(
8484
image_uri: str,
8585
inference_spec: InferenceSpec = None,
8686
) -> str:
87-
"""This is a one-line summary of the function.
87+
"""Prepares for InferenceSpec using model_path, writes inference.py, and captures dependencies to generate secret_key.
8888
Args:to
8989
model_path (str) : Argument
9090
shared_libs (List[]) : Argument
@@ -93,7 +93,7 @@ def prepare_for_mms(
9393
inference_spec (InferenceSpec, optional) : Argument
9494
(default is None)
9595
Returns:
96-
( str ) :
96+
( str ) : secret_key
9797
"""
9898
model_path = Path(model_path)
9999
if not model_path.exists():

0 commit comments

Comments
 (0)