Skip to content

Commit 33dcf96

Browse files
committed
re-format lmi config switch
1 parent 8b96155 commit 33dcf96

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/sagemaker/serve/builder/model_builder.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1303,7 +1303,9 @@ def _model_builder_optimize_wrapper(
13031303
if self._is_jumpstart_model_id():
13041304
self.build(mode=self.mode, sagemaker_session=self.sagemaker_session)
13051305
if self.pysdk_model:
1306-
self.pysdk_model.set_deployment_config(instance_type=instance_type, config_name="lmi")
1306+
self.pysdk_model.set_deployment_config(
1307+
instance_type=instance_type, config_name="lmi"
1308+
)
13071309
input_args = self._optimize_for_jumpstart(
13081310
output_path=output_path,
13091311
instance_type=instance_type,

0 commit comments

Comments
 (0)