nit

huggingface · May 1, 2024 · eb65ff4 · eb65ff4
1 parent 3315804
commit eb65ff4
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/.circleci/create_circleci_config.py b/.circleci/create_circleci_config.py
@@ -229,7 +229,7 @@ def job_name(self):
     "torch",
     docker_image=[{"image": "huggingface/transformers-torch-light"}],
     install_steps=["uv venv && uv pip install ."],
-    parallelism=16,
+    parallelism=4,
     pytest_num_workers=16
 )
 
@@ -238,15 +238,17 @@ def job_name(self):
     "tf",
     docker_image=[{"image":"huggingface/transformers-tf-light"}],
     install_steps=["uv venv", "uv pip install -e."],
-    parallelism=8
+    pytest_num_workers=16,
+    parallelism=4
 )
 
 
 flax_job = CircleCIJob(
     "flax",
     docker_image=[{"image":"huggingface/transformers-jax-light"}],
     install_steps=["uv venv && uv pip install ."],
-    parallelism=8
+    parallelism=4,
+    pytest_num_workers=16
 )