pypi_publish.yaml

name: PyPI release

on:
  workflow_dispatch:
  push:
    tags:
      - "v*"

jobs:
  build-and-test:
    runs-on: ubuntu-22.04
    steps:
      - name: Checkout repo
        uses: actions/checkout@v6

      - name: Set up Python
        uses: actions/setup-python@v6
        with:
          python-version: "3.10"

      - name: Fetch and checkout latest release branch
        run: |
          pip install -U requests packaging
          LATEST_BRANCH=$(python utils/fetch_latest_release_branch.py)
          echo "Latest branch: $LATEST_BRANCH"
          git fetch origin "$LATEST_BRANCH"
          git checkout "$LATEST_BRANCH"

      - name: Install build dependencies
        run: |
          python -m pip install --upgrade pip
          pip install -U build
          pip install -U torch --index-url https://download.pytorch.org/whl/cpu

      - name: Build the dist files
        run: python -m build

      - name: Install from built wheel
        run: pip install dist/*.whl

      - name: Test installing diffusers and importing
        run: |
          pip install -U transformers
          python utils/print_env.py
          python -c "from diffusers import __version__; print(__version__)"
          python -c "from diffusers import DiffusionPipeline; pipe = DiffusionPipeline.from_pretrained('fusing/unet-ldm-dummy-update'); pipe()"
          python -c "from diffusers import DiffusionPipeline; pipe = DiffusionPipeline.from_pretrained('hf-internal-testing/tiny-stable-diffusion-pipe', safety_checker=None); pipe('ah suh du')"
          python -c "from diffusers import *"

      - name: Upload build artifacts
        uses: actions/upload-artifact@v4
        with:
          name: python-dist
          path: dist/

  publish-to-pypi:
    needs: build-and-test
    if: startsWith(github.ref, 'refs/tags/')
    runs-on: ubuntu-22.04
    environment: pypi-release
    permissions:
      id-token: write

    steps:
      - name: Download build artifacts
        uses: actions/download-artifact@v4
        with:
          name: python-dist
          path: dist/

      - name: Publish to PyPI
        uses: pypa/gh-action-pypi-publish@release/v1