update tgi container

huggingface · Oct 25, 2024 · 4815b6a · 4815b6a
1 parent 1de603e
commit 4815b6a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/source/containers.mdx b/docs/source/containers.mdx
@@ -34,7 +34,7 @@ print(f"llm image uri: {llm_image}")
 |-----------------------------|-----------------|---------------------------------------------|
 | Training  | 0.0.24           | `763104351884.dkr.ecr.us-west-2.amazonaws.com/huggingface-pytorch-training-neuronx:2.1.2-transformers4.41.1-neuronx-py310-sdk2.19.1-ubuntu20.04`   |
 | Inference      | 0.0.24           | `763104351884.dkr.ecr.us-west-2.amazonaws.com/huggingface-pytorch-inference-neuronx:2.1.2-transformers4.41.1-neuronx-py310-sdk2.19.1-ubuntu20.04`      |
-| Text Generation Inference        | 0.0.24           | `763104351884.dkr.ecr.us-west-2.amazonaws.com/huggingface-pytorch-tgi-inference:2.1.2-optimum0.0.24-neuronx-py310-ubuntu22.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`        |
 
 
 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.