From 6cffc90dcacca489f785fa4689addbc49f8dd17f Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Wed, 28 Aug 2024 09:05:52 +0200 Subject: [PATCH] nit --- docker/README.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/docker/README.md b/docker/README.md index 9b37e30a62da30..2a71ab6fb6ec8f 100644 --- a/docker/README.md +++ b/docker/README.md @@ -4,4 +4,6 @@ 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! :hugs: \ No newline at end of file