From b2cb734ef57713ea756152ec5efd25133bec072c Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Wed, 3 Apr 2024 11:36:55 +0200 Subject: [PATCH] low hanging fruit there --- .circleci/create_circleci_config.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/.circleci/create_circleci_config.py b/.circleci/create_circleci_config.py index 12a9c43fbd3db2..896fc091bae218 100644 --- a/.circleci/create_circleci_config.py +++ b/.circleci/create_circleci_config.py @@ -377,12 +377,12 @@ def job_name(self): hub_job = CircleCIJob( "hub", additional_env={"HUGGINGFACE_CO_STAGING": True}, + docker_image=[{"image":"huggingface/transformers-consistency"}], # TODO do we really need torch for that job? torch-light? + install_steps=["uv venv", "uv pip install -e ."], install_steps=[ "sudo apt-get -y update && sudo apt-get install git-lfs", 'git config --global user.email "ci@dummy.com"', 'git config --global user.name "ci"', - "pip install --upgrade --upgrade-strategy eager pip", - "pip install -U --upgrade-strategy eager .[torch,sentencepiece,testing,vision]", ], marker="is_staging_test", pytest_num_workers=1, @@ -419,10 +419,8 @@ def job_name(self): repo_utils_job = CircleCIJob( "repo_utils", - install_steps=[ - "pip install --upgrade --upgrade-strategy eager pip", - "pip install -U --upgrade-strategy eager .[quality,testing,torch]", - ], + docker_image=[{"image":"huggingface/transformers-consistency"}], # TODO do we really need torch for that job? torch-light? + install_steps=["uv venv", "uv pip install -e ."], parallelism=None, pytest_num_workers=1, resource_class="large",