Merge branch 'master' into update-pt-2.5.1

sage-maker · web-flow · commit eabac704363c · 2025-03-20T09:21:02.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
diff --git a/tests/integ/sagemaker/serve/test_schema_builder.py b/tests/integ/sagemaker/serve/test_schema_builder.py
@@ -34,7 +34,9 @@
 
 def test_model_builder_happy_path_with_only_model_id_text_generation(sagemaker_session):
     model_builder = ModelBuilder(
-        model="HuggingFaceH4/zephyr-7b-beta", sagemaker_session=sagemaker_session
+        model="HuggingFaceH4/zephyr-7b-beta",
+        sagemaker_session=sagemaker_session,
+        instance_type=None,
     )
 
     model = model_builder.build(sagemaker_session=sagemaker_session)

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`
Original file line number	Diff line number	Diff line change
`@@ -34,7 +34,9 @@`
`34`	`34`
`35`	`35`	`def test_model_builder_happy_path_with_only_model_id_text_generation(sagemaker_session):`
`36`	`36`	`model_builder = ModelBuilder(`
`37`		`- model="HuggingFaceH4/zephyr-7b-beta", sagemaker_session=sagemaker_session`
	`37`	`+ model="HuggingFaceH4/zephyr-7b-beta",`
	`38`	`+ sagemaker_session=sagemaker_session,`
	`39`	`+ instance_type=None,`
`38`	`40`	`)`
`39`	`41`
`40`	`42`	`model = model_builder.build(sagemaker_session=sagemaker_session)`