mirror of
https://github.com/ace-step/ACE-Step-1.5.git
synced 2026-07-02 16:37:04 +00:00
78 lines
2.4 KiB
Bash
Executable file
78 lines
2.4 KiB
Bash
Executable file
#!/usr/bin/env bash
|
|
# Test Environment Auto-Detection - Linux/macOS
|
|
# This script tests the environment detection logic
|
|
|
|
set -euo pipefail
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
|
|
echo "========================================"
|
|
echo "ACE-Step Environment Detection Test"
|
|
echo "========================================"
|
|
echo
|
|
|
|
OS_NAME="$(uname)"
|
|
ARCH="$(uname -m)"
|
|
echo "[Info] Platform: $OS_NAME ($ARCH)"
|
|
echo
|
|
|
|
# Test 1: Check for uv
|
|
echo "[Test 1] Checking for uv command..."
|
|
if command -v uv &>/dev/null; then
|
|
echo "[PASS] uv detected"
|
|
uv --version
|
|
elif [[ -x "$HOME/.local/bin/uv" ]]; then
|
|
echo "[PASS] uv detected at $HOME/.local/bin/uv (not in PATH)"
|
|
"$HOME/.local/bin/uv" --version
|
|
elif [[ -x "$HOME/.cargo/bin/uv" ]]; then
|
|
echo "[PASS] uv detected at $HOME/.cargo/bin/uv (not in PATH)"
|
|
"$HOME/.cargo/bin/uv" --version
|
|
else
|
|
echo "[INFO] uv not found"
|
|
echo "To install uv, run: curl -LsSf https://astral.sh/uv/install.sh | sh"
|
|
fi
|
|
echo
|
|
|
|
# Test 2: Check project.scripts in pyproject.toml
|
|
echo "[Test 2] Checking project scripts..."
|
|
if [[ -f "$SCRIPT_DIR/pyproject.toml" ]]; then
|
|
echo "[PASS] pyproject.toml found"
|
|
echo
|
|
echo "Available scripts:"
|
|
grep -E "^acestep" "$SCRIPT_DIR/pyproject.toml" 2>/dev/null | head -5 || echo " (not found)"
|
|
else
|
|
echo "[FAIL] pyproject.toml not found"
|
|
fi
|
|
echo
|
|
|
|
# Test 3: Check accelerator
|
|
echo "[Test 3] Checking accelerator..."
|
|
if [[ "$OS_NAME" == "Darwin" && "$ARCH" == "arm64" ]]; then
|
|
echo "[INFO] Apple Silicon - MLX backend available"
|
|
echo " Recommended: ./start_gradio_ui_macos.sh"
|
|
elif [[ "$OS_NAME" == "Linux" ]]; then
|
|
if command -v nvidia-smi &>/dev/null; then
|
|
echo "[INFO] NVIDIA CUDA GPU available"
|
|
echo " Recommended: ./start_gradio_ui.sh"
|
|
else
|
|
echo "[INFO] No NVIDIA GPU detected - CPU mode"
|
|
echo " Recommended: ./start_gradio_ui.sh"
|
|
fi
|
|
else
|
|
echo "[INFO] Platform: $OS_NAME $ARCH"
|
|
fi
|
|
echo
|
|
|
|
# Test 4: Environment selection logic
|
|
echo "[Test 4] Environment selection logic..."
|
|
if command -v uv &>/dev/null || [[ -x "$HOME/.local/bin/uv" ]] || [[ -x "$HOME/.cargo/bin/uv" ]]; then
|
|
echo "[RESULT] Will use: uv package manager"
|
|
echo "Command: uv run acestep"
|
|
else
|
|
echo "[ERROR] uv not found!"
|
|
echo "Please install uv first."
|
|
fi
|
|
echo
|
|
|
|
echo "========================================"
|
|
echo "Test Complete"
|
|
echo "========================================"
|