deep-learning-containers icon indicating copy to clipboard operation
deep-learning-containers copied to clipboard

[HuggingFace][Neuronx] DLC for Optimum-neuron 0.0.25 - Neuron SDK 2.20.0 PyTorch 2.1.2 - Transformers 4.43.2

Open dacorvo opened this issue 4 months ago • 0 comments

Overview of DLCs to update

Inference - Neuronx

Dependencies versions:

transformers: 4.43.2 torch: 2.1.2 aws-neuron-sdk: 2.20.0 optimum-neuron: 0.0.25

Training - Neuronx

Dependencies versions:

transformers: 4.43.2 torch: 2.1.2 aws-neuron-sdk: 2.20.0 optimum-neuron: 0.0.25

cc. @catalin-manciu-aws @philschmid @mmcclean-aws @Captainia @JingyaHuang

dacorvo avatar Oct 01 '24 13:10 dacorvo