build-ci-docker-images.yml

name: Build docker

on:
  push:
    branches:
      - change-ci
  repository_dispatch:
  workflow_call:
    inputs:
      image_postfix:
        required: true
        type: string
  schedule:
    - cron: "17 0 * * *"

concurrency:
  cancel-in-progress: true

jobs:
  build:
    # Push CI doesn't need this image
    # if: inputs.image_postfix != '-push-ci'
    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"]  # Add your file names here

    runs-on: ubuntu-22.04

    steps:
      -
        name: Set up Docker Buildx
        uses: docker/setup-buildx-action@v3
      -
        name: Check out code
        uses: actions/checkout@v3
      -
        name: Login to DockerHub
        uses: docker/login-action@v3
        with:
          username: ${{ secrets.DOCKERHUB_USERNAME }}
          password: ${{ secrets.DOCKERHUB_PASSWORD }}
      -
        name: ${{ matrix.file }}
        uses: docker/build-push-action@v5
        with:
          context: ./docker
          build-args: |
            REF=main
          file: "./docker/${{ matrix.file }}.dockerfile"
          push: true
          tags: huggingface/transformers-${{ matrix.file }}