diff --git a/.github/workflows/build-ci-docker-images.yml b/.github/workflows/build-ci-docker-images.yml index a82071dfa8e9dc..99fc9a26400a56 100644 --- a/.github/workflows/build-ci-docker-images.yml +++ b/.github/workflows/build-ci-docker-images.yml @@ -26,7 +26,7 @@ jobs: strategy: matrix: file: ["quality", "consistency", "custom-tokenizers", "torch-light", "tf-light", "exotic-models", "torch-tf-light"] # Add your file names here - + continue-on-error: true if: | contains(github.event.head_commit.message, '[push-ci-image]') && contains(github.actor, 'transformers') && diff --git a/docker/consistency.dockerfile b/docker/consistency.dockerfile index 1d8748fb608823..8c77d2dda4073d 100644 --- a/docker/consistency.dockerfile +++ b/docker/consistency.dockerfile @@ -8,6 +8,6 @@ RUN uv venv RUN uv pip install --no-cache-dir -U pip setuptools RUN uv pip install --no-cache-dir tensorflow-cpu tf-keras pytest RUN uv pip install --no-cache --upgrade 'torch' --index-url https://download.pytorch.org/whl/cpu -RUN pip install --no-cache-dir --upgrade GitPython "transformers[flax,quality]" +RUN uv pip install --no-cache-dir --upgrade GitPython "transformers[flax,quality]" RUN pip uninstall -y transformers RUN apt-get clean && rm -rf /var/lib/apt/lists/* \ No newline at end of file diff --git a/docker/exotic-models.dockerfile b/docker/exotic-models.dockerfile index a9d311c878681f..c688c293fe6674 100644 --- a/docker/exotic-models.dockerfile +++ b/docker/exotic-models.dockerfile @@ -19,7 +19,7 @@ RUN uv pip install --no-cache-dir "transformers[testing, vision,timm]" 'pip>=2 RUN git clone https://github.com/facebookresearch/detectron2.git RUN python3 -m pip install --no-cache-dir -e detectron2 RUN pip uninstall -y transformers -RUN apt install libgl1-mesa-glx libgl1 +RUN apt-get install -y libgl1-mesa-glx libgl1 RUN apt-get clean && rm -rf /var/lib/apt/lists/* RUN pip cache remove "nvidia-*" RUN pip cache remove triton \ No newline at end of file