fix: factor in set instance type when building JumpStart models in ModelBuilder.

Joseph Zhang · Joseph Zhang · commit 0a723566ca31 · 2025-03-18T17:46:33.000-07:00
diff --git a/src/sagemaker/serve/builder/jumpstart_builder.py b/src/sagemaker/serve/builder/jumpstart_builder.py
@@ -157,6 +157,7 @@ def _create_pre_trained_js_model(self) -> Type[Model]:
             vpc_config=self.vpc_config,
             sagemaker_session=self.sagemaker_session,
             name=self.name,
+            instance_type=self.instance_type,
         )
 
         self._original_deploy = pysdk_model.deploy

Original file line number	Diff line number	Diff line change
`@@ -157,6 +157,7 @@ def _create_pre_trained_js_model(self) -> Type[Model]:`
`157`	`157`	`vpc_config=self.vpc_config,`
`158`	`158`	`sagemaker_session=self.sagemaker_session,`
`159`	`159`	`name=self.name,`
	`160`	`+ instance_type=self.instance_type,`
`160`	`161`	`)`
`161`	`162`
`162`	`163`	`self._original_deploy = pysdk_model.deploy`