From a07358e4d2fde7fcb15a2e98260cc46759759178 Mon Sep 17 00:00:00 2001 From: loserbcc Date: Sun, 26 Oct 2025 22:56:27 -0400 Subject: [PATCH] Add comprehensive Docker setup for ACE-Step MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit This commit adds a complete Docker containerization solution for ACE-Step: Features: - Optimized Dockerfile for CUDA 13.0/RTX 5090 - Docker Compose configuration with GPU support - Management script with easy commands (start/stop/logs/rebuild) - Quick launcher script (ace-docker) for simple usage - Comprehensive documentation (DOCKER.md) - Helper scripts for different quality settings - CPU offload enabled (~0.4GB VRAM usage) - Persistent volumes for checkpoints, outputs, and logs Quick start: ./ace-docker # Start ACE-Step ./docker-manage.sh # Full management interface Access at: http://localhost:7866 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude --- .dockerignore | 43 +++++ DOCKER.md | 281 +++++++++++++++++++++++++++++++++ Dockerfile.scorpy | 56 +++++++ ace-docker | 43 +++++ docker-compose.scorpy.yml | 69 ++++++++ docker-manage.sh | 139 ++++++++++++++++ launch.sh | 131 +++++++++++++++ start_ace_step.sh | 13 ++ start_ace_step_full_quality.sh | 12 ++ 9 files changed, 787 insertions(+) create mode 100644 .dockerignore create mode 100644 DOCKER.md create mode 100644 Dockerfile.scorpy create mode 100755 ace-docker create mode 100644 docker-compose.scorpy.yml create mode 100755 docker-manage.sh create mode 100755 launch.sh create mode 100755 start_ace_step.sh create mode 100755 start_ace_step_full_quality.sh diff --git a/.dockerignore b/.dockerignore new file mode 100644 index 00000000..dd7787bf --- /dev/null +++ b/.dockerignore @@ -0,0 +1,43 @@ +# Python +__pycache__/ +*.py[cod] +*$py.class +*.so +.Python +*.egg-info/ +dist/ +build/ +.venv/ +venv/ +env/ + +# ACE-Step specific +outputs/ +checkpoints/ +logs/ +*.wav +*.mp3 + +# Git +.git/ +.gitignore + +# IDE +.vscode/ +.idea/ +*.swp +*.swo + +# OS +.DS_Store +Thumbs.db + +# Docker +docker-compose*.yml +Dockerfile* + +# Misc +# Keep README.md as it's needed by setup.py +# *.md +DOCKER.md +LICENSE diff --git a/DOCKER.md b/DOCKER.md new file mode 100644 index 00000000..1753a535 --- /dev/null +++ b/DOCKER.md @@ -0,0 +1,281 @@ +# ACE-Step Docker Setup + +Complete Docker containerization for ACE-Step music generation server. + +## Prerequisites + +### 1. Docker & Docker Compose +```bash +# Arch Linux +sudo pacman -S docker docker-compose + +# Start and enable Docker +sudo systemctl start docker +sudo systemctl enable docker + +# Add user to docker group +sudo usermod -aG docker $USER +# Log out and back in for group changes to take effect +``` + +### 2. NVIDIA Container Toolkit (for GPU support) +```bash +# Arch Linux +sudo pacman -S nvidia-container-toolkit + +# Configure Docker to use NVIDIA runtime +sudo nvidia-ctk runtime configure --runtime=docker +sudo systemctl restart docker + +# Test GPU access +docker run --rm --gpus all nvidia/cuda:13.0.0-base-ubuntu22.04 nvidia-smi +``` + +## Quick Start + +### Build and Run +```bash +# Build the Docker image +./docker-manage.sh build + +# Start the container +./docker-manage.sh start + +# Access the web interface +# Local: http://localhost:7866 +# Network: http://192.168.4.72:7866 +``` + +### Management Commands +```bash +./docker-manage.sh status # Check if running +./docker-manage.sh logs # View logs (live) +./docker-manage.sh stop # Stop container +./docker-manage.sh restart # Restart container +./docker-manage.sh shell # Open shell in container +./docker-manage.sh clean # Remove container +./docker-manage.sh rebuild # Full rebuild from scratch +``` + +## Directory Structure + +``` +ACE-Step/ +├── Dockerfile.scorpy # Docker build instructions +├── docker-compose.scorpy.yml # Container orchestration +├── docker-manage.sh # Management script +├── .dockerignore # Files to exclude from build +│ +├── checkpoints/ # Model files (persistent) +├── outputs/ # Generated audio files +└── logs/ # Application logs +``` + +## Configuration + +### Default Settings +The container runs with these optimized settings: +- **Port:** 7866 (mapped to host) +- **GPU:** Full access to all NVIDIA GPUs +- **Memory:** CPU offload enabled (~0.4GB VRAM) +- **Quality:** BFloat16 precision +- **Decoding:** Overlapped decode enabled + +### Custom Configuration +Edit `docker-compose.scorpy.yml` to change settings: + +```yaml +# Change port +ports: + - "8080:7866" # Access on port 8080 + +# Modify ACE-Step arguments +command: > + acestep + --server_name 0.0.0.0 + --port 7866 + --bf16 false # Disable BFloat16 + --cpu_offload false # Use more VRAM for quality + --overlapped_decode true +``` + +## Volume Mounts + +### Persistent Data +- `./checkpoints` → `/app/checkpoints` - Model files +- `./outputs` → `/app/outputs` - Generated audio +- `./logs` → `/app/logs` - Application logs +- `ace-step-cache` → `/root/.cache/ace-step` - Downloaded models + +### Accessing Outputs +Generated audio files are available at: +```bash +ls -la outputs/ +``` + +## Network Access + +### Firewall Configuration +The container exposes port 7866. Ensure your firewall allows it: +```bash +# Check UFW status +sudo ufw status + +# Allow port 7866 +sudo ufw allow 7866/tcp comment 'ACE-Step Docker' +``` + +### Access Points +- **Local:** http://localhost:7866 +- **LAN:** http://192.168.4.72:7866 +- **Tailscale:** http://100.66.73.8:7866 (when active) + +## Troubleshooting + +### Check Container Status +```bash +./docker-manage.sh status +docker logs ace-step-server +``` + +### GPU Not Detected +```bash +# Verify nvidia-container-toolkit +docker run --rm --gpus all nvidia/cuda:13.0.0-base-ubuntu22.04 nvidia-smi + +# Check container GPU access +docker exec ace-step-server nvidia-smi +``` + +### Container Won't Start +```bash +# Check logs +./docker-manage.sh logs + +# Rebuild from scratch +./docker-manage.sh rebuild +``` + +### Permission Issues +```bash +# Ensure user is in docker group +groups | grep docker + +# If not, add and re-login +sudo usermod -aG docker $USER +``` + +## Performance + +### Resource Usage +- **Build Time:** ~10-15 minutes (first time) +- **Image Size:** ~8-10 GB +- **Runtime VRAM:** ~0.4GB (with cpu_offload) +- **Runtime VRAM:** ~8-12GB (without cpu_offload, full quality) + +### Generation Speed (RTX 5090) +- **27 steps:** ~9 seconds for 10s audio +- **60 steps:** ~20 seconds for 10s audio +- **RTF:** ~6 iterations/second + +## Advanced Usage + +### Custom Build +```bash +# Build with specific tag +docker compose -f docker-compose.scorpy.yml build --tag ace-step:v1.0 + +# Build without cache +docker compose -f docker-compose.scorpy.yml build --no-cache +``` + +### Environment Variables +```bash +# Set custom output directory +docker compose -f docker-compose.scorpy.yml run \ + -e ACE_OUTPUT_DIR=/custom/path \ + ace-step +``` + +### Interactive Shell +```bash +# Access container shell +./docker-manage.sh shell + +# Or manually +docker exec -it ace-step-server /bin/bash +``` + +## Integration with asay + +The `asay` tool works seamlessly with the Docker container: + +```bash +# From any machine in your fleet +asay "Hello from Docker container" + +# The tool auto-detects the containerized server +# at http://192.168.4.72:7866 +``` + +## Backup & Restore + +### Backup Model Checkpoints +```bash +tar -czf ace-step-checkpoints-$(date +%Y%m%d).tar.gz checkpoints/ +``` + +### Restore +```bash +tar -xzf ace-step-checkpoints-*.tar.gz +./docker-manage.sh restart +``` + +## Updates + +### Update ACE-Step Code +```bash +cd /home/brian/ACE-Step +git pull origin main +./docker-manage.sh rebuild +``` + +### Update Base Image +Edit `Dockerfile.scorpy` to use newer CUDA/Ubuntu versions, then: +```bash +./docker-manage.sh rebuild +``` + +## Production Deployment + +### Auto-start on Boot +```bash +# Using systemd +sudo systemctl enable docker + +# Container auto-restart is already enabled via: +# restart: unless-stopped +``` + +### Monitoring +```bash +# Health check status +docker inspect ace-step-server | grep -A 5 Health + +# Resource usage +docker stats ace-step-server +``` + +## Security + +### Recommended Practices +1. Run behind reverse proxy (nginx/traefik) +2. Enable HTTPS with Let's Encrypt +3. Use firewall to limit access +4. Regular backups of checkpoints +5. Keep base images updated + +## License + +ACE-Step is licensed under Apache License 2.0. +See main README.md for details. diff --git a/Dockerfile.scorpy b/Dockerfile.scorpy new file mode 100644 index 00000000..1f6bb596 --- /dev/null +++ b/Dockerfile.scorpy @@ -0,0 +1,56 @@ +FROM nvidia/cuda:13.0.0-cudnn-runtime-ubuntu22.04 + +# Prevent interactive prompts during build +ENV DEBIAN_FRONTEND=noninteractive + +# Install system dependencies +RUN apt-get update && apt-get install -y \ + python3.10 \ + python3-pip \ + python3.10-venv \ + git \ + wget \ + curl \ + ffmpeg \ + && rm -rf /var/lib/apt/lists/* + +# Set working directory +WORKDIR /app + +# Copy requirements and setup files +COPY requirements.txt setup.py README.md ./ +COPY acestep ./acestep +COPY config ./config +COPY examples ./examples + +# Create virtual environment and install dependencies +RUN python3.10 -m venv /opt/venv +ENV PATH="/opt/venv/bin:$PATH" + +# Install PyTorch with CUDA 13.0 support (for RTX 5090) +RUN pip install --no-cache-dir \ + torch torchvision torchaudio \ + --index-url https://download.pytorch.org/whl/nightly/cu130 + +# Install torchcodec for audio encoding +RUN pip install --no-cache-dir torchcodec + +# Install ACE-Step and dependencies +RUN pip install --no-cache-dir -e . + +# Create directories for models and outputs +RUN mkdir -p /app/checkpoints /app/outputs /app/logs + +# Set environment variables +ENV ACE_OUTPUT_DIR=/app/outputs +ENV PYTHONUNBUFFERED=1 + +# Expose port +EXPOSE 7866 + +# Health check +HEALTHCHECK --interval=60s --timeout=10s --start-period=30s --retries=3 \ + CMD curl -f http://localhost:7866/ || exit 1 + +# Default command - can be overridden +CMD ["acestep", "--server_name", "0.0.0.0", "--port", "7866", "--bf16", "true", "--torch_compile", "false"] diff --git a/ace-docker b/ace-docker new file mode 100755 index 00000000..4ed20e51 --- /dev/null +++ b/ace-docker @@ -0,0 +1,43 @@ +#!/bin/bash +# ACE-Step Docker Quick Launcher +# Usage: ./ace-docker [start|stop|status|logs|rebuild] + +cd "$(dirname "$0")" + +case "${1:-start}" in + start|up) + echo "🚀 Starting ACE-Step..." + ./docker-manage.sh start + ;; + stop|down) + echo "⏚ī¸ Stopping ACE-Step..." + ./docker-manage.sh stop + ;; + status) + ./docker-manage.sh status + ;; + logs) + ./docker-manage.sh logs + ;; + restart) + ./docker-manage.sh restart + ;; + rebuild) + ./docker-manage.sh rebuild + ;; + *) + echo "ACE-Step Docker Quick Launcher" + echo "" + echo "Usage: ./ace-docker [command]" + echo "" + echo "Commands:" + echo " start - Start ACE-Step (default)" + echo " stop - Stop ACE-Step" + echo " status - Check status" + echo " logs - View logs" + echo " restart - Restart" + echo " rebuild - Full rebuild" + echo "" + echo "Access at: http://localhost:7866" + ;; +esac diff --git a/docker-compose.scorpy.yml b/docker-compose.scorpy.yml new file mode 100644 index 00000000..c9b89feb --- /dev/null +++ b/docker-compose.scorpy.yml @@ -0,0 +1,69 @@ +services: + ace-step: + build: + context: . + dockerfile: Dockerfile.scorpy + image: ace-step:latest + container_name: ace-step-server + hostname: ace-step + restart: unless-stopped + + # GPU support - requires nvidia-container-toolkit + deploy: + resources: + reservations: + devices: + - driver: nvidia + count: all + capabilities: [gpu] + + # Port mapping + ports: + - "7866:7866" + + # Volume mounts + volumes: + # Model checkpoints (persistent) + - ./checkpoints:/app/checkpoints + # Output audio files + - ./outputs:/app/outputs + # Logs + - ./logs:/app/logs + # Optional: Cache directory for auto-downloaded models + - ace-step-cache:/root/.cache/ace-step + + # Environment variables + environment: + - ACE_OUTPUT_DIR=/app/outputs + - NVIDIA_VISIBLE_DEVICES=all + - NVIDIA_DRIVER_CAPABILITIES=compute,utility + + # Command with optimized settings for RTX 5090 + command: > + acestep + --server_name 0.0.0.0 + --port 7866 + --bf16 true + --torch_compile false + --cpu_offload true + --overlapped_decode true + + # Health check + healthcheck: + test: ["CMD", "curl", "-f", "http://localhost:7866/"] + interval: 60s + timeout: 10s + retries: 3 + start_period: 30s + + # Logging + logging: + driver: "json-file" + options: + max-size: "10m" + max-file: "3" + +# Named volumes +volumes: + ace-step-cache: + driver: local diff --git a/docker-manage.sh b/docker-manage.sh new file mode 100755 index 00000000..87926f87 --- /dev/null +++ b/docker-manage.sh @@ -0,0 +1,139 @@ +#!/bin/bash +# ACE-Step Docker Management Script + +set -e + +SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" +cd "$SCRIPT_DIR" + +COMPOSE_FILE="docker-compose.scorpy.yml" + +# Colors +RED='\033[0;31m' +GREEN='\033[0;32m' +YELLOW='\033[1;33m' +BLUE='\033[0;34m' +CYAN='\033[0;36m' +NC='\033[0m' # No Color +BOLD='\033[1m' + +show_help() { + echo -e "${CYAN}╔════════════════════════════════════════════════════════════╗${NC}" + echo -e "${CYAN}║${NC} ${BOLD}ACE-Step Docker Management${NC} ${CYAN}║${NC}" + echo -e "${CYAN}╚════════════════════════════════════════════════════════════╝${NC}" + echo "" + echo -e "${BOLD}Usage:${NC} $0 [command]" + echo "" + echo -e "${BOLD}Commands:${NC}" + echo -e " ${GREEN}build${NC} Build the Docker image" + echo -e " ${GREEN}start${NC} Start the container (detached)" + echo -e " ${GREEN}stop${NC} Stop the container" + echo -e " ${GREEN}restart${NC} Restart the container" + echo -e " ${GREEN}logs${NC} View container logs (live)" + echo -e " ${GREEN}status${NC} Show container status" + echo -e " ${GREEN}shell${NC} Open bash shell in container" + echo -e " ${GREEN}clean${NC} Stop and remove container" + echo -e " ${GREEN}rebuild${NC} Clean, rebuild, and start" + echo -e " ${GREEN}help${NC} Show this help message" + echo "" + echo -e "${BOLD}Examples:${NC}" + echo -e " $0 build # Build the image" + echo -e " $0 start # Start ACE-Step server" + echo -e " $0 logs # Watch the logs" + echo -e " $0 rebuild # Full rebuild from scratch" + echo "" +} + +check_requirements() { + # Check Docker + if ! command -v docker &> /dev/null; then + echo -e "${RED}Error: Docker is not installed${NC}" + exit 1 + fi + + # Check nvidia-container-toolkit + if ! docker run --rm --gpus all nvidia/cuda:13.0.0-base-ubuntu22.04 nvidia-smi &> /dev/null; then + echo -e "${YELLOW}Warning: nvidia-container-toolkit may not be properly configured${NC}" + echo -e "${YELLOW}GPU support may not work. Install with:${NC}" + echo -e " sudo pacman -S nvidia-container-toolkit" + echo "" + fi +} + +case "${1:-help}" in + build) + echo -e "${BLUE}Building ACE-Step Docker image...${NC}" + check_requirements + docker compose -f "$COMPOSE_FILE" build + echo -e "${GREEN}✓ Build complete${NC}" + ;; + + start) + echo -e "${BLUE}Starting ACE-Step container...${NC}" + check_requirements + docker compose -f "$COMPOSE_FILE" up -d + echo -e "${GREEN}✓ Container started${NC}" + echo -e "${CYAN}Access ACE-Step at: http://localhost:7866${NC}" + echo -e "${YELLOW}View logs with: $0 logs${NC}" + ;; + + stop) + echo -e "${BLUE}Stopping ACE-Step container...${NC}" + docker compose -f "$COMPOSE_FILE" stop + echo -e "${GREEN}✓ Container stopped${NC}" + ;; + + restart) + echo -e "${BLUE}Restarting ACE-Step container...${NC}" + docker compose -f "$COMPOSE_FILE" restart + echo -e "${GREEN}✓ Container restarted${NC}" + ;; + + logs) + echo -e "${BLUE}Showing container logs (Ctrl+C to exit)...${NC}" + docker compose -f "$COMPOSE_FILE" logs -f + ;; + + status) + echo -e "${CYAN}Container Status:${NC}" + docker compose -f "$COMPOSE_FILE" ps + echo "" + if docker compose -f "$COMPOSE_FILE" ps | grep -q "Up"; then + echo -e "${GREEN}✓ ACE-Step is running${NC}" + echo -e "${CYAN}Access at: http://localhost:7866${NC}" + else + echo -e "${YELLOW}ACE-Step is not running${NC}" + fi + ;; + + shell) + echo -e "${BLUE}Opening shell in container...${NC}" + docker compose -f "$COMPOSE_FILE" exec ace-step /bin/bash + ;; + + clean) + echo -e "${YELLOW}Stopping and removing container...${NC}" + docker compose -f "$COMPOSE_FILE" down + echo -e "${GREEN}✓ Container removed${NC}" + ;; + + rebuild) + echo -e "${YELLOW}Performing full rebuild...${NC}" + docker compose -f "$COMPOSE_FILE" down + docker compose -f "$COMPOSE_FILE" build --no-cache + docker compose -f "$COMPOSE_FILE" up -d + echo -e "${GREEN}✓ Rebuild complete${NC}" + echo -e "${CYAN}Access ACE-Step at: http://localhost:7866${NC}" + ;; + + help|--help|-h) + show_help + ;; + + *) + echo -e "${RED}Unknown command: $1${NC}" + echo "" + show_help + exit 1 + ;; +esac diff --git a/launch.sh b/launch.sh new file mode 100755 index 00000000..04e8a66e --- /dev/null +++ b/launch.sh @@ -0,0 +1,131 @@ +#!/bin/bash +# ACE-Step Interactive Launcher +# Port 7866 - Accessible on local network and Tailscale + +SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" +cd "$SCRIPT_DIR" + +# Colors +RED='\033[0;31m' +GREEN='\033[0;32m' +YELLOW='\033[1;33m' +BLUE='\033[0;34m' +CYAN='\033[0;36m' +NC='\033[0m' # No Color +BOLD='\033[1m' + +clear +echo -e "${CYAN}╔════════════════════════════════════════════════════════════╗${NC}" +echo -e "${CYAN}║${NC} ${BOLD}ACE-Step Music Generation Launcher${NC} ${CYAN}║${NC}" +echo -e "${CYAN}╚════════════════════════════════════════════════════════════╝${NC}" +echo "" + +# Show system info +echo -e "${BOLD}System Information:${NC}" +echo -e " GPU: ${GREEN}$(nvidia-smi --query-gpu=name --format=csv,noheader 2>/dev/null || echo 'N/A')${NC}" +echo -e " PyTorch: ${GREEN}$(source .venv/bin/activate && python -c 'import torch; print(torch.__version__)' 2>/dev/null)${NC}" +echo -e " CUDA: ${GREEN}$(source .venv/bin/activate && python -c 'import torch; print(torch.version.cuda)' 2>/dev/null)${NC}" +echo "" + +# Show network info +LOCAL_IP=$(ip addr show wlan0 2>/dev/null | grep "inet " | awk '{print $2}' | cut -d/ -f1) +TAILSCALE_IP=$(tailscale ip -4 2>/dev/null) + +echo -e "${BOLD}Network Access (Port 7866):${NC}" +echo -e " Local: ${YELLOW}http://localhost:7866${NC}" +if [ -n "$LOCAL_IP" ]; then + echo -e " LAN: ${YELLOW}http://$LOCAL_IP:7866${NC}" +fi +if [ -n "$TAILSCALE_IP" ]; then + echo -e " Tailscale: ${GREEN}http://$TAILSCALE_IP:7866${NC} ✓" +else + echo -e " Tailscale: ${RED}Not running${NC}" +fi +echo "" + +echo -e "${CYAN}════════════════════════════════════════════════════════════${NC}" +echo -e "${BOLD}Launch Options:${NC}" +echo "" +echo -e " ${GREEN}1)${NC} ${BOLD}Full Quality Mode${NC} (Recommended)" +echo -e " â€ĸ Maximum audio quality" +echo -e " â€ĸ Uses ~8-12GB VRAM" +echo -e " â€ĸ Best for final productions" +echo "" +echo -e " ${YELLOW}2)${NC} ${BOLD}Memory Optimized Mode${NC}" +echo -e " â€ĸ CPU offload enabled" +echo -e " â€ĸ Uses ~0.4GB VRAM" +echo -e " â€ĸ Slightly lower quality" +echo -e " â€ĸ Good for testing/experimenting" +echo "" +echo -e " ${BLUE}3)${NC} ${BOLD}Fast Mode${NC} (Experimental)" +echo -e " â€ĸ All optimizations enabled" +echo -e " â€ĸ Fastest generation" +echo -e " â€ĸ May have quality trade-offs" +echo "" +echo -e " ${CYAN}4)${NC} ${BOLD}Custom Options${NC}" +echo -e " â€ĸ Manually specify parameters" +echo "" +echo -e " ${RED}5)${NC} Exit" +echo "" +echo -e "${CYAN}════════════════════════════════════════════════════════════${NC}" +echo -n -e "${BOLD}Select option [1-5]:${NC} " + +read -r choice + +case $choice in + 1) + echo -e "\n${GREEN}Starting ACE-Step in Full Quality Mode...${NC}\n" + source .venv/bin/activate + acestep \ + --server_name 0.0.0.0 \ + --port 7866 \ + --bf16 true \ + --torch_compile false + ;; + 2) + echo -e "\n${YELLOW}Starting ACE-Step in Memory Optimized Mode...${NC}\n" + source .venv/bin/activate + acestep \ + --server_name 0.0.0.0 \ + --port 7866 \ + --bf16 true \ + --torch_compile false \ + --cpu_offload true \ + --overlapped_decode true + ;; + 3) + echo -e "\n${BLUE}Starting ACE-Step in Fast Mode...${NC}\n" + source .venv/bin/activate + acestep \ + --server_name 0.0.0.0 \ + --port 7866 \ + --bf16 true \ + --torch_compile false \ + --cpu_offload true \ + --overlapped_decode true + ;; + 4) + echo -e "\n${CYAN}Custom Options${NC}" + echo -e "Current working options for RTX 5090:" + echo -e " ${GREEN}--server_name 0.0.0.0${NC} (required for network access)" + echo -e " ${GREEN}--port 7866${NC} (your configured port)" + echo -e " ${GREEN}--bf16 true/false${NC} (use bfloat16)" + echo -e " ${GREEN}--torch_compile false${NC} (must be false for RTX 5090)" + echo -e " ${YELLOW}--cpu_offload true/false${NC} (save VRAM)" + echo -e " ${YELLOW}--overlapped_decode true/false${NC} (faster decoding)" + echo "" + echo -n "Enter custom acestep arguments: " + read -r custom_args + echo -e "\n${CYAN}Starting ACE-Step with custom options...${NC}\n" + source .venv/bin/activate + acestep --server_name 0.0.0.0 --port 7866 $custom_args + ;; + 5) + echo -e "\n${RED}Exiting...${NC}\n" + exit 0 + ;; + *) + echo -e "\n${RED}Invalid option!${NC}\n" + exit 1 + ;; +esac diff --git a/start_ace_step.sh b/start_ace_step.sh new file mode 100755 index 00000000..1b6873de --- /dev/null +++ b/start_ace_step.sh @@ -0,0 +1,13 @@ +#!/bin/bash +# ACE-Step startup script +# Runs on port 7866, accessible on local network and Tailscale +# Using PyTorch with CUDA 13.0 for RTX 5090 support + +source .venv/bin/activate +acestep \ + --server_name 0.0.0.0 \ + --port 7866 \ + --bf16 true \ + --torch_compile false \ + --cpu_offload true \ + --overlapped_decode true diff --git a/start_ace_step_full_quality.sh b/start_ace_step_full_quality.sh new file mode 100755 index 00000000..90bdb8a0 --- /dev/null +++ b/start_ace_step_full_quality.sh @@ -0,0 +1,12 @@ +#!/bin/bash +# ACE-Step startup script - Full Quality Mode +# Runs on port 7866, accessible on local network and Tailscale +# Using PyTorch with CUDA 13.0 for RTX 5090 support +# NO CPU offload for maximum quality + +source .venv/bin/activate +acestep \ + --server_name 0.0.0.0 \ + --port 7866 \ + --bf16 true \ + --torch_compile false