From f6d01d93d5627e5c7bf3224b629daff459179e81 Mon Sep 17 00:00:00 2001 From: ydshieh Date: Tue, 2 Jul 2024 15:39:44 +0200 Subject: [PATCH] torch 2.4 --- .circleci/create_circleci_config.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/.circleci/create_circleci_config.py b/.circleci/create_circleci_config.py index d6d1041ea656ad..fa07b6b2f16c91 100644 --- a/.circleci/create_circleci_config.py +++ b/.circleci/create_circleci_config.py @@ -246,7 +246,7 @@ def job_name(self): torch_job = CircleCIJob( "torch", docker_image=[{"image": "huggingface/transformers-torch-light"}], - install_steps=["uv venv && uv pip install ."], + install_steps=["uv venv && uv pip install .", "pip uninstall -y torch torchvision torchaudio", "pip install torch==2.4.0 torch==0.19.0 torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/test/cpu"], parallelism=6, pytest_num_workers=16 ) @@ -282,7 +282,7 @@ def job_name(self): "pipelines_torch", additional_env={"RUN_PIPELINE_TESTS": True}, docker_image=[{"image":"huggingface/transformers-torch-light"}], - install_steps=["uv venv && uv pip install ."], + install_steps=["uv venv && uv pip install .", "pip uninstall -y torch torchvision torchaudio", "pip install torch==2.4.0 torch==0.19.0 torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/test/cpu"], marker="is_pipeline_test", ) @@ -317,7 +317,7 @@ def job_name(self): cache_name="torch_examples", docker_image=[{"image":"huggingface/transformers-examples-torch"}], # TODO @ArthurZucker remove this once docker is easier to build - install_steps=["uv venv && uv pip install . && uv pip install -r examples/pytorch/_tests_requirements.txt"], + install_steps=["uv venv && uv pip install . && uv pip install -r examples/pytorch/_tests_requirements.txt", "pip uninstall -y torch torchvision torchaudio", "pip install torch==2.4.0 torch==0.19.0 torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/test/cpu"], pytest_num_workers=1, )