You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
System information
A description of your system. Please provide:
SageMaker Python SDK version: 2.212
Framework name (eg. PyTorch) or algorithm (eg. KMeans): PyTorch
Framework version: -
Python version: 3.10
CPU or GPU: GPU and CPU
Custom Docker image (Y/N): N
Additional context
SageMaker endpoint was working for a while and successfully processing requests. The endpoint restarted and installed the latest version of SageMaker SDK (2.212). The endpoint stopped processing requests and printed logs as above. I have noticed that ModelBuilder creates a model package with requirements.txt. In that file, there is sagemaker>=2.199. I modified it and set sagemaker==2.199 which solved the issue.
The text was updated successfully, but these errors were encountered:
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",
],
}
)
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"
Describe the bug
Model cannot be loaded in the SageMaker endpoint after update of SageMaker SDK to 2.212
To reproduce
Expected behavior
ModelBuilder
set==
not>=
2.212
Screenshots or logs
System information
A description of your system. Please provide:
Additional context
SageMaker endpoint was working for a while and successfully processing requests. The endpoint restarted and installed the latest version of SageMaker SDK (
2.212
). The endpoint stopped processing requests and printed logs as above. I have noticed thatModelBuilder
creates a model package withrequirements.txt
. In that file, there issagemaker>=2.199
. I modified it and setsagemaker==2.199
which solved the issue.The text was updated successfully, but these errors were encountered: