update [push-ci-image]

.github/workflows/build-ci-docker-images.yml

@@ -1,4 +1,4 @@
-name: Build docker
+name: Build pr ci-docker
 
 on:
   push:
@@ -17,23 +17,20 @@ on:
 
 jobs:
   build:
-    # Push CI doesn't need this image
-    # if: inputs.image_postfix != '-push-ci'
-
     runs-on: ubuntu-22.04
     concurrency:
       group: staging_environment
       cancel-in-progress: true
-    if: |
-      contains(github.event.head_commit.message, '[push-ci-image]')  && 
-      contains(github.actor, 'transformers') && 
-      ${{ !cancelled() && (success() || failure()) }}
 
 
     strategy:
       matrix:
         file: ["quality", "consistency", "custom-tokenizers", "torch-light", "tf-light", "exotic-models", "torch-tf-light"]  # Add your file names here
 
+    if: |
+      contains(github.event.head_commit.message, '[push-ci-image]')  && 
+      contains(github.actor, 'transformers') && 
+      ${{ !cancelled() && (success() || failure()) }}
 
     steps:
       -
@@ -49,7 +46,7 @@ jobs:
           username: ${{ secrets.DOCKERHUB_USERNAME }}
           password: ${{ secrets.DOCKERHUB_PASSWORD }}
       -
-        name: ${{ matrix.file }}
+        name: Build ${{ matrix.file }}.dockerfile
         uses: docker/build-push-action@v5
         with:
           context: ./docker

docker/torch-tf-light.dockerfile

@@ -14,6 +14,5 @@ RUN uv pip install --no-cache-dir "transformers[sklearn,torch,testing,sentencepi
 
 RUN pip uninstall -y transformers
 RUN apt-get clean && rm -rf /var/lib/apt/lists/*
-RUN apt-get autoremove  --purge -y cmake g++
 RUN pip cache remove "nvidia-*"
 RUN pip cache remove triton