Skip to content

Commit

Permalink
new [push-ci-image]
Browse files Browse the repository at this point in the history
  • Loading branch information
ArthurZucker committed Apr 3, 2024
1 parent 9afaa62 commit 3cccbaf
Show file tree
Hide file tree
Showing 2 changed files with 22 additions and 3 deletions.
6 changes: 3 additions & 3 deletions .circleci/config.yml
Original file line number Diff line number Diff line change
Expand Up @@ -97,7 +97,7 @@ jobs:
fetch_all_tests:
working_directory: ~/transformers
docker:
- image: huggingface/transformers-check-large
- image: huggingface/transformers-consistency
parallelism: 1
steps:
- checkout
Expand All @@ -121,7 +121,7 @@ jobs:
check_code_quality:
working_directory: ~/transformers
docker:
- image: huggingface/transformers-check-quality
- image: huggingface/transformers-consistency
resource_class: large
environment:
TRANSFORMERS_IS_CI: yes
Expand All @@ -145,7 +145,7 @@ jobs:
check_repository_consistency:
working_directory: ~/transformers
docker:
- image: huggingface/transformers-check-large
- image: huggingface/transformers-consistency
resource_class: large
environment:
TRANSFORMERS_IS_CI: yes
Expand Down
19 changes: 19 additions & 0 deletions docker/torch-tf-light.dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
FROM huggingface/transformers-tf-light
ENV PYTHONDONTWRITEBYTECODE=1
USER root
RUN apt-get update && apt-get install -y time git pkg-config make
ENV VIRTUAL_ENV=/usr/local
RUN pip install uv
RUN uv venv
RUN uv pip install --no-cache-dir -U pip setuptools


RUN uv pip install --no-cache-dir --upgrade 'torch' --index-url https://download.pytorch.org/whl/cpu
RUN uv pip install --no-cache-dir "transformers[sklearn,torch,testing,sentencepiece,vision,timm,torch-speech]"


RUN pip uninstall -y transformers
RUN apt-get clean && rm -rf /var/lib/apt/lists/*
RUN apt-get autoremove --purge -y cmake g++
RUN pip cache remove "nvidia-*"
RUN pip cache remove triton

0 comments on commit 3cccbaf

Please sign in to comment.