Skip to content

Commit 87c525f

Browse files
authored
Initialize framework and framework version in post_init (#5424)
* Initialize framework and version in post_init * Fix model registry notebook * removing redundant condition check
1 parent f272de0 commit 87c525f

File tree

2 files changed

+16
-5
lines changed

2 files changed

+16
-5
lines changed

sagemaker-serve/src/sagemaker/serve/model_builder.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -489,6 +489,11 @@ def _initialize_defaults(self) -> None:
489489
self.container_config = "host"
490490
self.inference_recommender_job_results = None
491491
self.container_log_level = logging.INFO
492+
493+
if not hasattr(self, 'framework'):
494+
self.framework = None
495+
if not hasattr(self, 'framework_version'):
496+
self.framework_version = None
492497

493498
def _fetch_default_instance_type_for_custom_model(self) -> str:
494499
hosting_configs = self._fetch_hosting_configs_for_custom_model()

v3-examples/ml-ops-examples/v3-model-registry-example/v3-model-registry-example.ipynb

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -185,18 +185,24 @@
185185
"metadata": {},
186186
"outputs": [],
187187
"source": [
188+
"from sagemaker.core.resources import ModelPackage\n",
189+
"from sagemaker.serve.model_builder import ModelBuilder\n",
190+
"\n",
191+
"# Get the registered ModelPackage using the ARN as model_package_name\n",
192+
"model_package = ModelPackage.get(model_package_name=registered_model_package_arn)\n",
193+
"\n",
194+
"s3_model_data_uri = model_package.inference_specification.containers[0].model_data_url\n",
195+
"image_uri = model_package.inference_specification.containers[0].image\n",
188196
"\n",
189-
"# Use the model_package in section above\n",
190197
"model_builder = ModelBuilder(\n",
191-
" model_package_arn=registered_model_package_arn,\n",
198+
" s3_model_data_url=s3_model_data_uri,\n",
199+
" image_uri=image_uri,\n",
192200
" role_arn=role,\n",
193201
" sagemaker_session=sagemaker_session\n",
194202
")\n",
195203
"\n",
196-
"\n",
197-
"\n",
198204
"# Build the model\n",
199-
"model = model_builder.build(model_name=\"my-model-from-registry\")"
205+
"model = model_builder.build(model_name=\"my-model-from-registry\")\n"
200206
]
201207
}
202208
],

0 commit comments

Comments
 (0)