From 93aabc68d100edd910c11a880855ef6124b0154b Mon Sep 17 00:00:00 2001
From: pagezyhf <165770107+pagezyhf@users.noreply.github.com>
Date: Tue, 3 Dec 2024 10:24:12 +0100
Subject: [PATCH] Training container update (#745)

training container
---
 docs/source/containers.mdx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docs/source/containers.mdx b/docs/source/containers.mdx
index c76bdc375..74f7fc618 100644
--- a/docs/source/containers.mdx
+++ b/docs/source/containers.mdx
@@ -32,7 +32,7 @@ print(f"llm image uri: {llm_image}")
 
 | Type                       | Optimum Version | Image URI                                   |
 |-----------------------------|-----------------|---------------------------------------------|
-| 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`   |
+| 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`        |