diff --git a/docker/docker-compose_gpu.yml b/docker/docker-compose_gpu.yml index 8cdb219..ed2bd7b 100644 --- a/docker/docker-compose_gpu.yml +++ b/docker/docker-compose_gpu.yml @@ -54,7 +54,7 @@ services: ports: - "5000:5000" volumes: - - ./magic-pdf.json:/root/magic-pdf.json + - ./magic-pdf-gpu.json:/root/magic-pdf.json depends_on: mysql: condition: service_healthy diff --git a/docker/magic-pdf-gpu.json b/docker/magic-pdf-gpu.json new file mode 100644 index 0000000..50c4fed --- /dev/null +++ b/docker/magic-pdf-gpu.json @@ -0,0 +1,62 @@ +{ + "bucket_info": { + "bucket-name-1": [ + "ak", + "sk", + "endpoint" + ], + "bucket-name-2": [ + "ak", + "sk", + "endpoint" + ] + }, + "models-dir": "/root/.cache/huggingface/hub/models--opendatalab--PDF-Extract-Kit-1.0/snapshots/782e787d46ed9b52253af6c1f69cdfcc76583e8d/models", + "layoutreader-model-dir": "/root/.cache/huggingface/hub/models--hantian--layoutreader/snapshots/641226775a0878b1014a96ad01b9642915136853", + "device-mode": "cuda", + "layout-config": { + "model": "doclayout_yolo" + }, + "formula-config": { + "mfd_model": "yolo_v8_mfd", + "mfr_model": "unimernet_small", + "enable": true + }, + "table-config": { + "model": "rapid_table", + "sub_model": "slanet_plus", + "enable": true, + "max_time": 400 + }, + "latex-delimiter-config": { + "display": { + "left": "$$", + "right": "$$" + }, + "inline": { + "left": "$", + "right": "$" + } + }, + "llm-aided-config": { + "formula_aided": { + "api_key": "your_api_key", + "base_url": "https://dashscope.aliyuncs.com/compatible-mode/v1", + "model": "qwen2.5-7b-instruct", + "enable": false + }, + "text_aided": { + "api_key": "your_api_key", + "base_url": "https://dashscope.aliyuncs.com/compatible-mode/v1", + "model": "qwen2.5-7b-instruct", + "enable": false + }, + "title_aided": { + "api_key": "your_api_key", + "base_url": "https://dashscope.aliyuncs.com/compatible-mode/v1", + "model": "qwen2.5-32b-instruct", + "enable": false + } + }, + "config_version": "1.2.1" +} \ No newline at end of file