aws / aws/sagemaker-huggingface-inference-toolkit

Sagemaker endpoint inferencing error with HF model loading from s3bucket with new transformer update

Open
#108 5 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
270
Forks
60
PR merge metrics
No merged PRs in 30d

Description

Since the transformer update (4.35.0). model outputs model.safetensor by default instead of pytorch.bin

endpoint created in sagamaker with model saved in s3bucket is throwing error. asking to define `"HF_TASK"`.

As per documentation of toolkit we do not need to define env variables when loading model from s3 Bucket, just a customer inference.py is needed.

Does someone know what's going wrong here ? @philschmid any idea ?

Contributor guide

Open the contributing guide

Research direction

Start by reproducing the SageMaker endpoint setup with the model saved in the S3 bucket, using the customer inference.py mentioned in the report and the transformer 4.35.0 model output. Compare the toolkit documentation's S3-loading behavior with the endpoint error requesting HF_TASK; done means identifying and documenting the compatibility or configuration issue.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
cloud, machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.