File tree Expand file tree Collapse file tree 2 files changed +20
-0
lines changed Expand file tree Collapse file tree 2 files changed +20
-0
lines changed Original file line number Diff line number Diff line change @@ -534,6 +534,16 @@ def _validate_model_id_and_get_type_hook():
534
534
if not self .model_type and not hub_arn :
535
535
raise ValueError (INVALID_MODEL_ID_ERROR_MSG .format (model_id = model_id ))
536
536
537
+ self ._specs = verify_model_region_and_return_specs (
538
+ region = self .region ,
539
+ model_id = self .model_id ,
540
+ version = self .model_version ,
541
+ hub_arn = self .hub_arn ,
542
+ model_type = self .model_type ,
543
+ scope = JumpStartScriptScope .TRAINING ,
544
+ sagemaker_session = self .sagemaker_session ,
545
+ )
546
+
537
547
estimator_init_kwargs = get_init_kwargs (
538
548
model_id = model_id ,
539
549
model_version = model_version ,
Original file line number Diff line number Diff line change @@ -312,6 +312,16 @@ def _validate_model_id_and_type():
312
312
if not self .model_type and not hub_arn :
313
313
raise ValueError (INVALID_MODEL_ID_ERROR_MSG .format (model_id = model_id ))
314
314
315
+ self ._specs = verify_model_region_and_return_specs (
316
+ region = self .region ,
317
+ model_id = self .model_id ,
318
+ version = self .model_version ,
319
+ hub_arn = self .hub_arn ,
320
+ model_type = self .model_type ,
321
+ scope = JumpStartScriptScope .INFERENCE ,
322
+ sagemaker_session = self .sagemaker_session ,
323
+ )
324
+
315
325
self ._model_data_is_set = model_data is not None
316
326
model_init_kwargs = get_init_kwargs (
317
327
model_id = model_id ,
You can’t perform that action at this time.
0 commit comments