oupsy

huggingface · Apr 26, 2024 · 37755b7 · 37755b7
1 parent 13fef30
commit 37755b7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.circleci/create_circleci_config.py b/.circleci/create_circleci_config.py
@@ -248,7 +248,7 @@ def job_name(self):
 torch_and_flax_job = CircleCIJob(
     "torch_and_flax",
     additional_env={"RUN_PT_FLAX_CROSS_TESTS": True},
-    docker_image=[{"image":"huggingface/transformers-torch-flax-light"}],
+    docker_image=[{"image":"huggingface/transformers-torch-jax-light"}],
     install_steps=["uv venv", "uv pip install -e ."],
     marker="is_pt_flax_cross_test",
     pytest_options={"rA": None, "durations": 0},
@@ -273,7 +273,7 @@ def job_name(self):
 
 flax_job = CircleCIJob(
     "flax",
-    docker_image=[{"image":"huggingface/transformers-flax-light"}],
+    docker_image=[{"image":"huggingface/transformers-jax-light"}],
     install_steps=["uv venv", "uv pip install -e ."],
     parallelism=1,
 )