Skip to content

Commit 33e83d8

Browse files
author
Malav Shastri
committed
feat: Add support for Hub in model attach functionality
1 parent 5449eb5 commit 33e83d8

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/sagemaker/jumpstart/model.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -429,6 +429,7 @@ def attach(
429429
cls,
430430
endpoint_name: str,
431431
inference_component_name: Optional[str] = None,
432+
hub_name: Optional[str] = None,
432433
model_id: Optional[str] = None,
433434
model_version: Optional[str] = None,
434435
sagemaker_session=DEFAULT_JUMPSTART_SAGEMAKER_SESSION,
@@ -457,6 +458,7 @@ def attach(
457458
model_id=model_id,
458459
model_version=model_version,
459460
sagemaker_session=sagemaker_session,
461+
hub_name=hub_name,
460462
)
461463
model.endpoint_name = endpoint_name
462464
model.inference_component_name = inference_component_name

0 commit comments

Comments
 (0)