Skip to content

Commit ffd3178

Browse files
author
Joseph Zhang
committed
Reformat sharded model validations.
1 parent 1bddf72 commit ffd3178

File tree

1 file changed

+13
-13
lines changed

1 file changed

+13
-13
lines changed

src/sagemaker/model.py

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -1600,21 +1600,21 @@ def deploy(
16001600
if self._base_name is not None:
16011601
self._base_name = "-".join((self._base_name, compiled_model_suffix))
16021602

1603-
if self._is_sharded_model and endpoint_type != EndpointType.INFERENCE_COMPONENT_BASED:
1604-
logging.warning(
1605-
"Forcing INFERENCE_COMPONENT_BASED endpoint for sharded model. ADVISORY - "
1606-
"Use INFERENCE_COMPONENT_BASED endpoints over MODEL_BASED endpoints."
1607-
)
1608-
endpoint_type = EndpointType.INFERENCE_COMPONENT_BASED
1603+
if self._is_sharded_model:
1604+
if endpoint_type != EndpointType.INFERENCE_COMPONENT_BASED:
1605+
logging.warning(
1606+
"Forcing INFERENCE_COMPONENT_BASED endpoint for sharded model. ADVISORY - "
1607+
"Use INFERENCE_COMPONENT_BASED endpoints over MODEL_BASED endpoints."
1608+
)
1609+
endpoint_type = EndpointType.INFERENCE_COMPONENT_BASED
16091610

1610-
if self._is_sharded_model and self._enable_network_isolation:
1611-
raise ValueError(
1612-
"EnableNetworkIsolation cannot be set to True since SageMaker Fast Model "
1613-
"Loading of model requires network access."
1614-
)
1611+
if self._enable_network_isolation:
1612+
raise ValueError(
1613+
"EnableNetworkIsolation cannot be set to True since SageMaker Fast Model "
1614+
"Loading of model requires network access."
1615+
)
16151616

1616-
if self._is_sharded_model:
1617-
if resources.num_cpus and resources.num_cpus > 0:
1617+
if resources and resources.num_cpus and resources.num_cpus > 0:
16181618
logger.warning(
16191619
"NumberOfCpuCoresRequired should be 0 for the best experience with SageMaker "
16201620
"Fast Model Loading. Configure by setting `num_cpus` to 0 in `resources`."

0 commit comments

Comments
 (0)