Skip to content

Files

Latest commit

9dc1efa · Feb 5, 2025

History

History
This branch is 129 commits behind huggingface/transformers:main.

docker

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
Jan 31, 2025
Jan 11, 2024
Sep 12, 2022
Feb 5, 2025
Jan 28, 2025
Jan 28, 2025
Jan 29, 2025
Feb 5, 2025
Jan 31, 2025
Mar 25, 2022
Dec 23, 2024
Oct 28, 2024
Oct 14, 2024
Feb 4, 2025
Feb 4, 2025
Feb 4, 2025
Feb 4, 2025
Feb 4, 2025
Feb 4, 2025
Feb 4, 2025
Feb 4, 2025
Feb 4, 2025
Feb 4, 2025
Feb 4, 2025
Feb 4, 2025
Feb 4, 2025

Dockers for transformers

In this folder you will find various docker files, and some subfolders.

  • dockerfiles (ex: consistency.dockerfile) present under ~/docker are used for our "fast" CIs. You should be able to use them for tasks that only need CPU. For example torch-light is a very light weights container (703MiB).
  • subfloder contain dockerfiles used for our slow CIs, which can be used for GPU tasks, but they are BIG as they were not specifically designed for a single model / single task. Thus the ~/docker/transformers-pytorch-gpu includes additional dependencies to allow us to run ALL model tests (say librosa or tesseract, which you do not need to run LLMs)

Note that in both case, you need to run uv pip install -e ., which should take around 5 seconds. We do it outside the dockerfile for the need of our CI: we checkout a new branch each time, and the transformers code is thus updated.

We are open to contribution, and invite the community to create dockerfiles with potential arguments that properly choose extras depending on the model's dependencies! 🤗