use torch-light for hub

huggingface · May 2, 2024 · 8dda4c3 · 8dda4c3
1 parent 900f779
commit 8dda4c3
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 15 deletions.
diff --git a/.circleci/create_circleci_config.py b/.circleci/create_circleci_config.py
@@ -308,7 +308,7 @@ def job_name(self):
 hub_job = CircleCIJob(
     "hub",
     additional_env={"HUGGINGFACE_CO_STAGING": True},
-    docker_image=[{"image":"huggingface/transformers-hub"}],
+    docker_image=[{"image":"huggingface/transformers-torch-light"}],
     install_steps=[
         "uv venv && uv pip install .",
         'git config --global user.email "[email protected]"',

diff --git a/.github/workflows/build-ci-docker-images.yml b/.github/workflows/build-ci-docker-images.yml
@@ -26,7 +26,7 @@ jobs:
 
     strategy:
       matrix:
-        file: ["quality", "consistency", "custom-tokenizers", "torch-light", "tf-light", "exotic-models", "torch-tf-light", "torch-jax-light", "jax-light", "hub"]
+        file: ["quality", "consistency", "custom-tokenizers", "torch-light", "tf-light", "exotic-models", "torch-tf-light", "torch-jax-light", "jax-light"]
     continue-on-error: true 
 
     steps:

diff --git a/docker/hub.dockerfile b/docker/hub.dockerfile