34 lines
1.3 KiB
Bash
34 lines
1.3 KiB
Bash
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
NETWORK=${NETWORK:-unis-ai}
|
|
API_NAME=${API_NAME:-mineru-api}
|
|
WEB_NAME=${WEB_NAME:-mindmap-web}
|
|
WEB_PORT=${WEB_PORT:-9898}
|
|
MINERU_API_URL=${MINERU_API_URL:-http://${API_NAME}:8000}
|
|
MINERU_GPU_MEMORY_UTILIZATION=${MINERU_GPU_MEMORY_UTILIZATION:-0.5}
|
|
MINERU_MODEL_DIR=${MINERU_MODEL_DIR:-/data/unis/mineru-models}
|
|
MINERU_PYTHONPATH=${MINERU_PYTHONPATH:-/app:/usr/local/corex-4.4.0/lib64/python3/dist-packages}
|
|
|
|
sudo docker network create "$NETWORK" >/dev/null 2>&1 || true
|
|
sudo docker rm -f "$API_NAME" "$WEB_NAME" >/dev/null 2>&1 || true
|
|
sudo mkdir -p "$MINERU_MODEL_DIR"/pipeline "$MINERU_MODEL_DIR"/vlm
|
|
sudo chmod -R 777 "$MINERU_MODEL_DIR"
|
|
|
|
sudo docker run -d --name "$API_NAME" --network "$NETWORK" \
|
|
--privileged --ipc=host --shm-size=32g \
|
|
-v "${MINERU_MODEL_DIR}:/models" \
|
|
-e MINERU_MODEL_SOURCE="${MINERU_MODEL_SOURCE:-local}" \
|
|
-e MINERU_MODEL_DIR=/models \
|
|
-e MINERU_PIPELINE_MODEL_DIR=/models/pipeline \
|
|
-e MINERU_VLM_MODEL_DIR=/models/vlm \
|
|
-e PYTHONPATH="$MINERU_PYTHONPATH" \
|
|
-e MINERU_GPU_MEMORY_UTILIZATION="$MINERU_GPU_MEMORY_UTILIZATION" \
|
|
unis-mindmap-api:latest \
|
|
mineru-api --host 0.0.0.0 --port 8000
|
|
|
|
sudo docker run -d --name "$WEB_NAME" --network "$NETWORK" \
|
|
-p "${WEB_PORT}:9898" \
|
|
-e MINERU_API_URL="$MINERU_API_URL" \
|
|
unis-mindmap-web:latest
|