From 5198d99e75915792ad32453d0b8061c844ecd3a9 Mon Sep 17 00:00:00 2001
From: zstar <65890619+zstar1003@users.noreply.github.com>
Date: Tue, 27 May 2025 10:17:30 +0800
Subject: [PATCH] =?UTF-8?q?refactor(docker):=20=E6=9B=B4=E6=96=B0GPU?=
 =?UTF-8?q?=E7=89=88docker-compose=E9=85=8D=E7=BD=AE=EF=BC=8C=E6=97=A0?=
 =?UTF-8?q?=E9=9C=80=E5=86=8D=E6=AC=A1=E4=BF=AE=E6=94=B9magic-pdf=E9=85=8D?=
 =?UTF-8?q?=E7=BD=AE=E6=96=87=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 docker/docker-compose_gpu.yml |  2 +-
 docker/magic-pdf-gpu.json     | 62 +++++++++++++++++++++++++++++++++++
 2 files changed, 63 insertions(+), 1 deletion(-)
 create mode 100644 docker/magic-pdf-gpu.json

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