Skip to content

Commit

Permalink
more fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
ArthurZucker committed Apr 26, 2024
1 parent db5cda3 commit abca599
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/build-ci-docker-images.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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') &&
Expand Down
2 changes: 1 addition & 1 deletion docker/consistency.dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -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/*
2 changes: 1 addition & 1 deletion docker/exotic-models.dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -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

0 comments on commit abca599

Please sign in to comment.