torch 2.4

huggingface · Jul 2, 2024 · f6d01d9 · f6d01d9
1 parent a970195
commit f6d01d9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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,
 )