Benson Wong
916d13f5bd
Add Docker build scripts for a unified cuda docker container with llama-server, stable-diffusion.cpp, whisper.cpp.
203 lines
6.6 KiB
Bash
Executable File
203 lines
6.6 KiB
Bash
Executable File
#!/bin/bash
|
|
#
|
|
# Build script for unified CUDA container with version pinning
|
|
#
|
|
# Usage:
|
|
# ./build-image.sh # Build with auto-detected versions
|
|
# ./build-image.sh --no-cache # Build without cache
|
|
# LLAMA_REF=b1234 ./build-image.sh # Pin llama.cpp to a commit hash
|
|
# LLAMA_REF=v1.2.3 ./build-image.sh # Pin llama.cpp to a tag
|
|
# LLAMA_REF=my-branch ./build-image.sh # Pin llama.cpp to a branch
|
|
# WHISPER_REF=v1.0.0 ./build-image.sh # Pin whisper.cpp to a tag
|
|
# SD_REF=master ./build-image.sh # Pin stable-diffusion.cpp to a branch
|
|
# LS_VERSION=170 ./build-image.sh # Override llama-swap version
|
|
#
|
|
|
|
set -euo pipefail
|
|
|
|
NO_CACHE=false
|
|
|
|
for arg in "$@"; do
|
|
case $arg in
|
|
--no-cache)
|
|
NO_CACHE=true
|
|
;;
|
|
--help|-h)
|
|
echo "Usage: ./build-image.sh [--no-cache]"
|
|
echo ""
|
|
echo "Environment variables:"
|
|
echo " DOCKER_IMAGE_TAG Set custom image tag (default: llama-swap:unified)"
|
|
echo " LLAMA_REF Pin llama.cpp to a commit, tag, or branch"
|
|
echo " WHISPER_REF Pin whisper.cpp to a commit, tag, or branch"
|
|
echo " SD_REF Pin stable-diffusion.cpp to a commit, tag, or branch"
|
|
echo " LS_VERSION Override llama-swap version (e.g., '170' or 'latest')"
|
|
exit 0
|
|
;;
|
|
esac
|
|
done
|
|
|
|
DOCKER_IMAGE_TAG="${DOCKER_IMAGE_TAG:-llama-swap:unified}"
|
|
|
|
# Git repository URLs
|
|
LLAMA_REPO="https://github.com/ggml-org/llama.cpp.git"
|
|
WHISPER_REPO="https://github.com/ggml-org/whisper.cpp.git"
|
|
SD_REPO="https://github.com/leejet/stable-diffusion.cpp.git"
|
|
|
|
# Resolve a git ref (commit hash, tag, or branch) to a full commit hash.
|
|
# Requires only: git, network access to the remote.
|
|
resolve_ref() {
|
|
local repo_url="$1"
|
|
local ref="$2"
|
|
|
|
# Full 40-char SHA — use as-is
|
|
if [[ "${ref}" =~ ^[0-9a-f]{40}$ ]]; then
|
|
echo "${ref}"
|
|
return
|
|
fi
|
|
|
|
# Try tag then branch (exact match)
|
|
local hash
|
|
hash=$(git ls-remote "${repo_url}" "refs/tags/${ref}" "refs/heads/${ref}" 2>/dev/null | head -1 | cut -f1)
|
|
if [[ -n "${hash}" ]]; then
|
|
echo "${hash}"
|
|
return
|
|
fi
|
|
|
|
# Short hash (7+ chars): scan all refs for a SHA with this prefix
|
|
if [[ "${ref}" =~ ^[0-9a-f]{7,}$ ]]; then
|
|
hash=$(git ls-remote "${repo_url}" 2>/dev/null | grep "^${ref}" | head -1 | cut -f1)
|
|
if [[ -n "${hash}" ]]; then
|
|
echo "${hash}"
|
|
return
|
|
fi
|
|
fi
|
|
|
|
echo "ERROR: Could not resolve ref '${ref}' for ${repo_url}" >&2
|
|
if [[ "${ref}" =~ ^[0-9a-f]+$ && ${#ref} -lt 7 ]]; then
|
|
echo " Short hashes must be at least 7 characters (got ${#ref})." >&2
|
|
else
|
|
echo " Tried: tag, branch, git ls-remote prefix match" >&2
|
|
fi
|
|
echo " Use a full 40-char SHA, a tag name, a branch name, or a 7+ char short hash." >&2
|
|
return 1
|
|
}
|
|
|
|
# Resolve HEAD of a repo without needing to know the default branch name.
|
|
get_latest_hash() {
|
|
git ls-remote "${1}" HEAD 2>/dev/null | head -1 | cut -f1
|
|
}
|
|
|
|
echo "=========================================="
|
|
echo "llama-swap Unified CUDA Build"
|
|
echo "=========================================="
|
|
echo ""
|
|
|
|
# Resolve llama.cpp ref
|
|
if [[ -n "${LLAMA_REF:-}" ]]; then
|
|
LLAMA_HASH=$(resolve_ref "${LLAMA_REPO}" "${LLAMA_REF}") || exit 1
|
|
echo "llama.cpp: ${LLAMA_REF} -> ${LLAMA_HASH}"
|
|
else
|
|
LLAMA_HASH=$(get_latest_hash "${LLAMA_REPO}")
|
|
if [[ -z "${LLAMA_HASH}" ]]; then
|
|
echo "ERROR: Could not determine latest commit for llama.cpp" >&2
|
|
exit 1
|
|
fi
|
|
echo "llama.cpp: latest HEAD: ${LLAMA_HASH}"
|
|
fi
|
|
|
|
# Resolve whisper.cpp ref
|
|
if [[ -n "${WHISPER_REF:-}" ]]; then
|
|
WHISPER_HASH=$(resolve_ref "${WHISPER_REPO}" "${WHISPER_REF}") || exit 1
|
|
echo "whisper.cpp: ${WHISPER_REF} -> ${WHISPER_HASH}"
|
|
else
|
|
WHISPER_HASH=$(get_latest_hash "${WHISPER_REPO}")
|
|
if [[ -z "${WHISPER_HASH}" ]]; then
|
|
echo "ERROR: Could not determine latest commit for whisper.cpp" >&2
|
|
exit 1
|
|
fi
|
|
echo "whisper.cpp: latest HEAD: ${WHISPER_HASH}"
|
|
fi
|
|
|
|
# Resolve stable-diffusion.cpp ref
|
|
if [[ -n "${SD_REF:-}" ]]; then
|
|
SD_HASH=$(resolve_ref "${SD_REPO}" "${SD_REF}") || exit 1
|
|
echo "stable-diffusion.cpp: ${SD_REF} -> ${SD_HASH}"
|
|
else
|
|
SD_HASH=$(get_latest_hash "${SD_REPO}")
|
|
if [[ -z "${SD_HASH}" ]]; then
|
|
echo "ERROR: Could not determine latest commit for stable-diffusion.cpp" >&2
|
|
exit 1
|
|
fi
|
|
echo "stable-diffusion.cpp: latest HEAD: ${SD_HASH}"
|
|
fi
|
|
|
|
# Resolve llama-swap version
|
|
LS_VER="${LS_VERSION:-latest}"
|
|
echo "llama-swap: ${LS_VER}"
|
|
|
|
echo ""
|
|
echo "=========================================="
|
|
echo "Starting Docker build..."
|
|
echo "=========================================="
|
|
echo ""
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
|
|
BUILD_ARGS=(
|
|
--build-arg "LLAMA_COMMIT_HASH=${LLAMA_HASH}"
|
|
--build-arg "WHISPER_COMMIT_HASH=${WHISPER_HASH}"
|
|
--build-arg "SD_COMMIT_HASH=${SD_HASH}"
|
|
--build-arg "LS_VERSION=${LS_VER}"
|
|
-t "${DOCKER_IMAGE_TAG}"
|
|
-f "${SCRIPT_DIR}/Dockerfile"
|
|
)
|
|
|
|
if [[ "$NO_CACHE" == true ]]; then
|
|
BUILD_ARGS+=(--no-cache)
|
|
echo "Note: Building without cache"
|
|
fi
|
|
|
|
DOCKER_BUILDKIT=1 docker buildx build --load "${BUILD_ARGS[@]}" "${SCRIPT_DIR}"
|
|
|
|
echo ""
|
|
echo "=========================================="
|
|
echo "Verifying build artifacts..."
|
|
echo "=========================================="
|
|
echo ""
|
|
|
|
MISSING_BINARIES=()
|
|
for binary in llama-server llama-cli whisper-server whisper-cli sd-server sd-cli llama-swap; do
|
|
if ! docker run --rm "${DOCKER_IMAGE_TAG}" which "${binary}" >/dev/null 2>&1; then
|
|
MISSING_BINARIES+=("${binary}")
|
|
fi
|
|
done
|
|
|
|
if [[ ${#MISSING_BINARIES[@]} -gt 0 ]]; then
|
|
echo "ERROR: Build succeeded but the following binaries are missing:"
|
|
for binary in "${MISSING_BINARIES[@]}"; do
|
|
echo " - ${binary}"
|
|
done
|
|
echo ""
|
|
echo "Try running with --no-cache flag:"
|
|
echo " ./build-image.sh --no-cache"
|
|
exit 1
|
|
fi
|
|
|
|
echo "All expected binaries verified: llama-server, llama-cli, whisper-server, whisper-cli, sd-server, sd-cli, llama-swap"
|
|
|
|
echo ""
|
|
echo "=========================================="
|
|
echo "Build complete!"
|
|
echo "=========================================="
|
|
echo ""
|
|
echo "Image tag: ${DOCKER_IMAGE_TAG}"
|
|
echo ""
|
|
echo "Built with:"
|
|
echo " llama.cpp: ${LLAMA_HASH}"
|
|
echo " whisper.cpp: ${WHISPER_HASH}"
|
|
echo " stable-diffusion.cpp: ${SD_HASH}"
|
|
echo " llama-swap: $(docker run --rm "${DOCKER_IMAGE_TAG}" cat /versions.txt | grep llama-swap | cut -d' ' -f2-)"
|
|
echo ""
|
|
echo "Run with:"
|
|
echo " docker run -it --rm --gpus all ${DOCKER_IMAGE_TAG}"
|