Skip to content

Commit 0675647

Browse files
authored
[Sagemaker] update Neuron DLCs (#1992)
* on training dlc * inf on dlc * on tgi dlc
1 parent f9dfe2c commit 0675647

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

docs/sagemaker/source/dlcs/available.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ You can also keep track of the latest Pytorch Training DLC releases [here](https
1515
| Container URI | Accelerator |
1616
| -------------------------------------------------------------------------------------------------------------------------------- | ----------- |
1717
| 763104351884.dkr.ecr.us-east-1.amazonaws.com/huggingface-pytorch-training:2.5.1-transformers4.49.0-gpu-py311-cu124-ubuntu22.04 | GPU |
18-
| 763104351884.dkr.ecr.us-east-1.amazonaws.com/huggingface-pytorch-training-neuronx:2.1.2-transformers4.48.1-neuronx-py310-sdk2.20.0-ubuntu20.04 | Neuron |
18+
| 763104351884.dkr.ecr.us-west-2.amazonaws.com/huggingface-pytorch-training-neuronx:2.7.0-transformers4.51.0-neuronx-py310-sdk2.24.1-ubuntu22.04 | Neuron |
1919

2020
## Inference
2121

@@ -29,7 +29,7 @@ You can also keep track of the latest Pytorch Inference DLC releases [here](http
2929
| -------------------------------------------------------------------------------------------------------------------------------- | ----------- |
3030
| 763104351884.dkr.ecr.us-east-1.amazonaws.com/huggingface-pytorch-inference:2.6.0-transformers4.51.3-cpu-py312-ubuntu22.04- | CPU |
3131
| 763104351884.dkr.ecr.us-east-1.amazonaws.com/huggingface-pytorch-inference:2.6.0-transformers4.51.3-gpu-py312-cu124-ubuntu22.04 | GPU |
32-
| 763104351884.dkr.ecr.us-east-1.amazonaws.com/huggingface-pytorch-inference-neuronx:2.1.2-transformers4.43.2-neuronx-py310-sdk2.20.0-ubuntu20.04 | Neuron |
32+
| 763104351884.dkr.ecr.us-west-2.amazonaws.com/huggingface-pytorch-inference-neuronx:2.7.1-transformers4.51.3-neuronx-py310-sdk2.24.1-ubuntu22.04 | Neuron |
3333

3434
### LLM TGI
3535

@@ -40,7 +40,7 @@ You can also keep track of the latest LLM TGI DLC releases [here](https://github
4040
| Container URI | Accelerator |
4141
| -------------------------------------------------------------------------------------------------------------------------------- | ----------- |
4242
| 763104351884.dkr.ecr.us-east-1.amazonaws.com/huggingface-pytorch-tgi-inference:2.7.0-tgi3.3.4-gpu-py311-cu124-ubuntu22.04 | GPU |
43-
| 763104351884.dkr.ecr.us-east-1.amazonaws.com/huggingface-pytorch-tgi-inference:2.5.1-optimum3.3.4-neuronx-py310-ubuntu22.04 | Neuron |
43+
| 763104351884.dkr.ecr.us-west-2.amazonaws.com/huggingface-pytorch-tgi-inference:2.7.0-optimum3.3.6-neuronx-py310-ubuntu22.04 | Neuron |
4444

4545
### Text Embedding Inference
4646

0 commit comments

Comments
 (0)