Benson Wong
e5e7391b6d
Update docker/unified scripts to support building both cuda and vulkan unified images.
90 lines
3.0 KiB
YAML
90 lines
3.0 KiB
YAML
name: Build Unified Docker Image
|
|
|
|
on:
|
|
schedule:
|
|
- cron: "37 5 * * *"
|
|
|
|
workflow_dispatch:
|
|
inputs:
|
|
llama_cpp_ref:
|
|
description: "llama.cpp commit hash, tag, or branch"
|
|
required: false
|
|
default: "master"
|
|
whisper_ref:
|
|
description: "whisper.cpp commit hash, tag, or branch"
|
|
required: false
|
|
default: "master"
|
|
sd_ref:
|
|
description: "stable-diffusion.cpp commit hash, tag, or branch"
|
|
required: false
|
|
default: "master"
|
|
llama_swap_version:
|
|
description: "llama-swap version (e.g. v198, latest, main)"
|
|
required: false
|
|
default: "main"
|
|
|
|
permissions:
|
|
contents: read
|
|
packages: write
|
|
|
|
jobs:
|
|
build:
|
|
runs-on: ubuntu-latest
|
|
strategy:
|
|
fail-fast: false
|
|
matrix:
|
|
backend: [cuda, vulkan]
|
|
steps:
|
|
- name: Checkout code
|
|
uses: actions/checkout@v4
|
|
|
|
- name: Free up disk space
|
|
run: |
|
|
echo "Before cleanup:"
|
|
df -h
|
|
sudo rm -rf /usr/share/dotnet
|
|
sudo rm -rf /usr/local/lib/android
|
|
sudo rm -rf /opt/ghc
|
|
sudo rm -rf /opt/hostedtoolcache/CodeQL
|
|
sudo docker system prune -af
|
|
echo "After cleanup:"
|
|
df -h
|
|
|
|
# On GitHub Actions runners, create a fresh builder.
|
|
# When running locally under act, skip this and reuse the existing
|
|
# llama-swap-builder (which has ccache warm) to avoid exhausting disk.
|
|
- name: Set up Docker Buildx
|
|
if: ${{ !env.ACT }}
|
|
uses: docker/setup-buildx-action@v3
|
|
|
|
- name: Log in to GitHub Container Registry
|
|
if: ${{ !env.ACT }}
|
|
uses: docker/login-action@v3
|
|
with:
|
|
registry: ghcr.io
|
|
username: ${{ github.actor }}
|
|
password: ${{ secrets.GITHUB_TOKEN }}
|
|
|
|
- name: Build unified Docker image (${{ matrix.backend }})
|
|
env:
|
|
LLAMA_REF: ${{ inputs.llama_cpp_ref || 'master' }}
|
|
WHISPER_REF: ${{ inputs.whisper_ref || 'master' }}
|
|
SD_REF: ${{ inputs.sd_ref || 'master' }}
|
|
LS_VERSION: ${{ inputs.llama_swap_version || 'main' }}
|
|
DOCKER_IMAGE_TAG: ghcr.io/mostlygeek/llama-swap:unified-${{ matrix.backend }}
|
|
# When running under act, use the local builder that has warm ccache.
|
|
# On GitHub Actions, BUILDX_BUILDER is unset so docker uses the builder
|
|
# created by setup-buildx-action above.
|
|
BUILDX_BUILDER: ${{ env.ACT == 'true' && 'llama-swap-builder' || '' }}
|
|
run: |
|
|
chmod +x docker/unified/build-image.sh
|
|
docker/unified/build-image.sh --${{ matrix.backend }}
|
|
|
|
- name: Push to GitHub Container Registry
|
|
if: ${{ !env.ACT }}
|
|
run: |
|
|
docker push ghcr.io/mostlygeek/llama-swap:unified-${{ matrix.backend }}
|
|
DATE_TAG=$(date -u +%Y-%m-%d)
|
|
docker tag ghcr.io/mostlygeek/llama-swap:unified-${{ matrix.backend }} ghcr.io/mostlygeek/llama-swap:unified-${{ matrix.backend }}-${DATE_TAG}
|
|
docker push ghcr.io/mostlygeek/llama-swap:unified-${{ matrix.backend }}-${DATE_TAG}
|