Model cannot be loaded in the SageMaker endpoint after update of SageMaker SDK to 2.212 about sagemaker-python-sdk

[samruds]

Taking a look. Will reproduce the error locally today.

[samruds]

Hello we have identified and fixed the problem to be related to an extra dependencies that was added to ModelBuilder.

Please pull in the latest commit of SDK if you are still seeing an issue with this version. Specifically pull in this commit #4549

[samruds]

Short term mitigations are

1. Pass a custom dependency

model_builder = ModelBuilder(
    #mode=Mode.SAGEMAKER_ENDPOINT,  # you can change it to Mode.LOCAL_CONTAINER for local testing
    mode=Mode.LOCAL_CONTAINER ,
    model_path=resnet_model_dir,
    inference_spec=my_inference_spec,
    schema_builder=my_schema,
    role_arn=execution_role,
    dependencies={
        "custom": [
            "accelerate==0.24.1",

        ],
    }
)

2. Installing the extras for bringing in accelerate if using the ModelBuilder interface. !pip install --force-reinstall --no-cache-dir --quiet "sagemaker[huggingface]>=2.212.0"

[samruds]

I will sync with SDK team on Monday for next steps to work with customer.