Skip to content

Add HF Transformers v5 training image#5695

Merged
Jyothirmaikottu merged 10 commits intoaws:masterfrom
ehcalabres:transformers-training-v5
Mar 17, 2026
Merged

Add HF Transformers v5 training image#5695
Jyothirmaikottu merged 10 commits intoaws:masterfrom
ehcalabres:transformers-training-v5

Conversation

@ehcalabres
Copy link
Copy Markdown

@ehcalabres ehcalabres commented Feb 24, 2026

Add new Transformers v5 image with updated dependencies:

  • Transformers: 5.3.0
  • Datasets: 4.5.0
  • HuggingFace Hub: 1.3.0
  • Diffusers: 0.37.0
  • TRL: 0.28.0
  • PEFT: 0.18.1
  • Trackio: 0.15.0

Purpose

Test Plan

Test Result


Toggle if you are merging into master Branch

By default, docker image builds and tests are disabled. Two ways to run builds and tests:

  1. Using dlc_developer_config.toml
  2. Using this PR description (currently only supported for PyTorch, TensorFlow, vllm, and base images)
How to use the helper utility for updating dlc_developer_config.toml

Assuming your remote is called origin (you can find out more with git remote -v)...

  • Run default builds and tests for a particular buildspec - also commits and pushes changes to remote; Example:

python src/prepare_dlc_dev_environment.py -b </path/to/buildspec.yml> -cp origin

  • Enable specific tests for a buildspec or set of buildspecs - also commits and pushes changes to remote; Example:

python src/prepare_dlc_dev_environment.py -b </path/to/buildspec.yml> -t sanity_tests -cp origin

  • Restore TOML file when ready to merge

python src/prepare_dlc_dev_environment.py -rcp origin

NOTE: If you are creating a PR for a new framework version, please ensure success of the local, standard, rc, and efa sagemaker tests by updating the dlc_developer_config.toml file:

  • sagemaker_remote_tests = true
  • sagemaker_efa_tests = true
  • sagemaker_rc_tests = true
  • sagemaker_local_tests = true
How to use PR description Use the code block below to uncomment commands and run the PR CodeBuild jobs. There are two commands available:
  • # /buildspec <buildspec_path>
    • e.g.: # /buildspec pytorch/training/buildspec.yml
    • If this line is commented out, dlc_developer_config.toml will be used.
  • # /tests <test_list>
    • e.g.: # /tests sanity security ec2
    • If this line is commented out, it will run the default set of tests (same as the defaults in dlc_developer_config.toml): sanity, security, ec2, ecs, eks, sagemaker, sagemaker-local.
# /buildspec <buildspec_path>
# /tests <test_list>
Toggle if you are merging into main Branch

PR Checklist

  • [] I ran pre-commit run --all-files locally before creating this PR. (Read DEVELOPMENT.md for details).

Add new Transformers v5 image with updated dependencies:
- Transformers: 5.1.0
- Datasets: 4.5.0
- HuggingFace Hub: 1.3.0
- Diffusers: 0.36.0
- TRL: 0.28.0
- PEFT: 0.18.1
- Trackio: 0.15.0
@ehcalabres ehcalabres marked this pull request as ready for review February 27, 2026 09:29
@ehcalabres ehcalabres requested a review from a team as a code owner February 27, 2026 09:29
Jyothirmaikottu
Jyothirmaikottu previously approved these changes Mar 6, 2026
Comment thread huggingface/pytorch/training/docker/2.9/py3/cu130/Dockerfile.gpu Outdated
Comment thread huggingface/pytorch/training/docker/2.9/py3/cu130/Dockerfile.gpu Outdated
Comment thread huggingface/pytorch/training/docker/2.9/py3/cu130/Dockerfile.gpu Outdated
Comment thread huggingface/pytorch/training/docker/2.9/py3/cu130/Dockerfile.gpu
Comment thread huggingface/pytorch/training/docker/2.9/py3/cu130/Dockerfile.gpu Outdated
Co-authored-by: Alvaro Bartolome <36760800+alvarobartt@users.noreply.github.com>
@Jyothirmaikottu Jyothirmaikottu merged commit a0652c9 into aws:master Mar 17, 2026
3 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants