File tree Expand file tree Collapse file tree 2 files changed +16
-5
lines changed
sagemaker-serve/src/sagemaker/serve
v3-examples/ml-ops-examples/v3-model-registry-example Expand file tree Collapse file tree 2 files changed +16
-5
lines changed Original file line number Diff line number Diff 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 ()
Original file line number Diff line number Diff line change 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 ],
You can’t perform that action at this time.
0 commit comments