Skip to content

[HuggingFace][Neuronx] Training - Optimum Neuron 0.1.0 - Neuron sdk 2.21.1 - Transformers to 4.48.3 #4670

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 15 commits into
base: master
Choose a base branch
from

Conversation

michaelbenayoun
Copy link

Description

This updates the HuggingFace Neuron training DLC to use AWS Neuron SDK 2.21.1 and optimum-neuron 0.1.0.

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license. I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

@michaelbenayoun michaelbenayoun requested a review from a team as a code owner March 27, 2025 15:03
@aws-deep-learning-containers-ci aws-deep-learning-containers-ci bot added authorized build Reflects file change in build folder huggingface Reflects file change in huggingface folder sagemaker_tests Size:S Determines the size of the PR test Reflects file change in test folder labels Apr 1, 2025
@michaelbenayoun michaelbenayoun changed the title [HuggingFace][Neuronx] Inference - Optimum Neuron 0.1.0 - Neuron sdk 2.21.1 - Transformers to 4.48.3 [HuggingFace][Neuronx] Training - Optimum Neuron 0.1.0 - Neuron sdk 2.21.1 - Transformers to 4.48.3 Apr 1, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
authorized build Reflects file change in build folder huggingface Reflects file change in huggingface folder sagemaker_tests Size:S Determines the size of the PR test Reflects file change in test folder unauthorized
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants