mirror of
https://github.com/mudler/LocalAI.git
synced 2026-04-01 13:42:20 -04:00
55 lines
1.6 KiB
Bash
Executable File
55 lines
1.6 KiB
Bash
Executable File
#!/bin/bash
|
|
set -e
|
|
|
|
CURDIR=$(dirname "$(realpath $0)")
|
|
|
|
echo "Running acestep-cpp backend tests..."
|
|
|
|
# The test requires:
|
|
# - ACESTEP_MODEL_DIR: path to directory containing GGUF model files
|
|
# - ACESTEP_BINARY: path to the acestep-cpp binary (defaults to ./acestep-cpp)
|
|
#
|
|
# Tests that require the model will be skipped if ACESTEP_MODEL_DIR is not set
|
|
# or the directory does not contain the required model files.
|
|
|
|
cd "$CURDIR"
|
|
|
|
# Only auto-download models when ACESTEP_MODEL_DIR is not explicitly set
|
|
if [ -z "$ACESTEP_MODEL_DIR" ]; then
|
|
export ACESTEP_MODEL_DIR="./acestep-models"
|
|
|
|
if [ ! -d "$ACESTEP_MODEL_DIR" ]; then
|
|
echo "Creating acestep-models directory for tests..."
|
|
mkdir -p "$ACESTEP_MODEL_DIR"
|
|
REPO_ID="Serveurperso/ACE-Step-1.5-GGUF"
|
|
echo "Repository: ${REPO_ID}"
|
|
echo ""
|
|
|
|
# Files to download (smallest quantizations for testing)
|
|
FILES=(
|
|
"acestep-5Hz-lm-0.6B-Q8_0.gguf"
|
|
"Qwen3-Embedding-0.6B-Q8_0.gguf"
|
|
"acestep-v15-turbo-Q8_0.gguf"
|
|
"vae-BF16.gguf"
|
|
)
|
|
|
|
BASE_URL="https://huggingface.co/${REPO_ID}/resolve/main"
|
|
|
|
for file in "${FILES[@]}"; do
|
|
dest="${ACESTEP_MODEL_DIR}/${file}"
|
|
if [ -f "${dest}" ]; then
|
|
echo " [skip] ${file} (already exists)"
|
|
else
|
|
echo " [download] ${file}..."
|
|
curl -L -o "${dest}" "${BASE_URL}/${file}" --progress-bar
|
|
echo " [done] ${file}"
|
|
fi
|
|
done
|
|
fi
|
|
fi
|
|
|
|
# Run Go tests
|
|
go test -v -timeout 600s .
|
|
|
|
echo "All acestep-cpp tests passed."
|