Skip to content

pass session to image_uris.retrieve() within huggingface get_llm_image_uri() func #4235

@Adamwgoh

Description

@Adamwgoh

Describe the bug
Currently image_uri retrieval always uses default jumpstart. For those who wants to use that function and passes a sagemaker session with profiles setup , it will not be received when running image_uris.retrieve(). the session is currently only used to retrieve the region of the session only

To reproduce
A clear, step-by-step set of instructions to reproduce the bug.

Expected behavior
sagemaker-python-sdk/src/sagemaker/huggingface

    if backend == "huggingface":
        return image_uris.retrieve(
            "huggingface-llm",
            region=region,
            version=version,
            image_scope="inference",
            sagemaker_session=session
        )

.

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions