aws / aws/amazon-sagemaker-examples-community

Hosting llama2 13b chat model using torch serve on inf2

Open
#9 1 comment 0 reactions 0 assignees View on GitHub
Dominant language
Jupyter Notebook
Stars
45
Forks
28
PR merge metrics
No merged PRs in 30d

Description

[1]https://github.com/aws/amazon-sagemaker-examples-community/blob/main/torchserve/inf2/llama2/llama-2-13b.ipynb

Was able to run the notebook successfully for llama2 13b.

Followed instructions to create artifacts for llama2 13b chat model, saved them in model_store/.., and ran inference to test.

While utilizing the notebook[1] to run the 13b chat model instead of the base model, getting a timeout error.
ReadTimeoutError: Read timeout on endpoint URL: "https://runtime.sagemaker.us-west-2.amazonaws.com/endpoints/.."

All other params are the same except tp_degree. I have used tp_degree=12 instead of 6 to utilize all cores on inf2.24x.

Contributor guide

Open the contributing guide

Research direction

Open the linked torchserve/inf2/llama2/llama-2-13b.ipynb and compare the successful base-model path with the 13b chat-model path, especially the tp_degree=12 configuration and endpoint invocation. Reproduce the SageMaker inference request, with successful completion without a ReadTimeoutError as the done condition.

Written by the indexing model from the issue text.

Assessment

Tech stack
aws, jupyter-notebook, pytorch
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.