diff --git a/docs/source/containers.mdx b/docs/source/containers.mdx index 74f7fc618..08e7dbdec 100644 --- a/docs/source/containers.mdx +++ b/docs/source/containers.mdx @@ -34,7 +34,7 @@ print(f"llm image uri: {llm_image}") |-----------------------------|-----------------|---------------------------------------------| | Training | 0.0.25 | `763104351884.dkr.ecr.us-west-2.amazonaws.com/huggingface-pytorch-training-neuronx:2.1.2-transformers4.43.2-neuronx-py310-sdk2.20.0-ubuntu20.04` | | Inference | 0.0.25 | `763104351884.dkr.ecr.us-west-2.amazonaws.com/huggingface-pytorch-inference-neuronx:2.1.2-transformers4.43.2-neuronx-py310-sdk2.20.0-ubuntu20.04` | -| Text Generation Inference | 0.0.25 | `763104351884.dkr.ecr.us-west-2.amazonaws.com/huggingface-pytorch-tgi-inference:2.1.2-optimum0.0.25-neuronx-py310-ubuntu22.04` | +| Text Generation Inference | 0.0.27 | `763104351884.dkr.ecr.us-west-2.amazonaws.com/huggingface-pytorch-tgi-inference:2.1.2-optimum0.0.27-neuronx-py310-ubuntu22.04` | Please replace `763104351884` with the correct [AWS account ID](https://github.com/aws/sagemaker-python-sdk/blob/master/src/sagemaker/image_uri_config/huggingface-neuronx.json) and `region` with the AWS region you are working in.