Ace-Step-Munk / run_gradio_training.sh
OnyxMunk's picture
Add LoRA training assets: scripts, docs (no binaries), ui, my_dataset
bc9c638
#!/usr/bin/env bash
# Launch Gradio with LoRA Training tab, low-VRAM friendly (6GB GPU).
# Use: ./run_gradio_training.sh
set -euo pipefail
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
cd "$SCRIPT_DIR"
export INIT_LLM="false"
export OFFLOAD_TO_CPU="true"
export PORT="${PORT:-7860}"
export SERVER_NAME="${SERVER_NAME:-127.0.0.1}"
export SHARE="${SHARE:-}"
source .venv/bin/activate
exec python -m acestep.acestep_v15_pipeline \
--port "$PORT" \
--server-name "$SERVER_NAME" \
--init_llm false \
--offload_to_cpu true \
--init_service true \
--config_path acestep-v15-turbo \
--download-source huggingface \
${SHARE:+--share} \
"$@"