diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml index 30a27bf..4af38b7 100644 --- a/docker/docker-compose.yml +++ b/docker/docker-compose.yml @@ -4,6 +4,7 @@ # ============================================================================= services: + # --- CLI 模式(默认)--- mineru: image: mineru-rocm:7.2.1 build: @@ -13,6 +14,7 @@ services: args: ARCH: gfx1201 GIT_PROXY: http://127.0.0.1:8118 + profiles: ["cli"] container_name: mineru-rocm stdin_open: true tty: true @@ -36,3 +38,61 @@ services: - ${MIOPEN_CACHE:-./data/miopen}:/root/.cache/miopen command: bash restart: "no" + + # --- WebUI(Gradio)--- + gradio: + image: mineru-rocm:7.2.1 + profiles: ["webui"] + container_name: mineru-gradio + stdin_open: true + tty: true + ipc: host + ports: + - "7860:7860" + devices: + - /dev/kfd + - /dev/dri + security_opt: + - seccomp=unconfined + group_add: + - video + - render + environment: + - MINERU_MODEL_SOURCE=${MINERU_MODEL_SOURCE:-huggingface} + - HF_HUB_CACHE=${HF_HUB_CACHE:-/opt/models/huggingface} + - MODELSCOPE_CACHE=${MODELSCOPE_CACHE:-/opt/models/modelscope} + volumes: + - ${INPUT_DIR:-./data/input}:/data/input:ro + - ${OUTPUT_DIR:-./data/output}:/data/output + - ${MODEL_DIR:-./data/models}:/opt/models + - ${MIOPEN_CACHE:-./data/miopen}:/root/.cache/miopen + command: mineru-gradio --server-name 0.0.0.0 --server-port 7860 + + # --- API 服务 --- + api: + image: mineru-rocm:7.2.1 + profiles: ["api"] + container_name: mineru-api + stdin_open: true + tty: true + ipc: host + ports: + - "8000:8000" + devices: + - /dev/kfd + - /dev/dri + security_opt: + - seccomp=unconfined + group_add: + - video + - render + environment: + - MINERU_MODEL_SOURCE=${MINERU_MODEL_SOURCE:-huggingface} + - HF_HUB_CACHE=${HF_HUB_CACHE:-/opt/models/huggingface} + - MODELSCOPE_CACHE=${MODELSCOPE_CACHE:-/opt/models/modelscope} + volumes: + - ${INPUT_DIR:-./data/input}:/data/input:ro + - ${OUTPUT_DIR:-./data/output}:/data/output + - ${MODEL_DIR:-./data/models}:/opt/models + - ${MIOPEN_CACHE:-./data/miopen}:/root/.cache/miopen + command: mineru-api --host 0.0.0.0 --port 8000