refactor(docker): 更新GPU版docker-compose配置,无需再次修改magic-pdf配置文件
This commit is contained in:
parent
cd44728c1a
commit
5198d99e75
|
@ -54,7 +54,7 @@ services:
|
||||||
ports:
|
ports:
|
||||||
- "5000:5000"
|
- "5000:5000"
|
||||||
volumes:
|
volumes:
|
||||||
- ./magic-pdf.json:/root/magic-pdf.json
|
- ./magic-pdf-gpu.json:/root/magic-pdf.json
|
||||||
depends_on:
|
depends_on:
|
||||||
mysql:
|
mysql:
|
||||||
condition: service_healthy
|
condition: service_healthy
|
||||||
|
|
|
@ -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"
|
||||||
|
}
|
Loading…
Reference in New Issue