diff --git a/docs/zh/usage/acceleration_cards/Kunlunxin.md b/docs/zh/usage/acceleration_cards/Kunlunxin.md
index 4bc19993..1d0861de 100644
--- a/docs/zh/usage/acceleration_cards/Kunlunxin.md
+++ b/docs/zh/usage/acceleration_cards/Kunlunxin.md
@@ -38,6 +38,7 @@ docker run -u root --name mineru_docker \
-v /usr/local/bin/xpu-smi:/usr/local/bin/xpu-smi \
-w /workspace \
-e MINERU_MODEL_SOURCE=local \
+ -e MINERU_FORMULA_CH_SUPPORT=true \
-e MINERU_VLLM_DEVICE=kxpu \
-it mineru:kxpu-vllm-latest \
/bin/bash
@@ -72,7 +73,7 @@ docker run -u root --name mineru_docker \
| <vlm/hybrid>-auto-engine |
- 🟡 |
+ 🟢 |
| <vlm/hybrid>-http-client |
@@ -85,7 +86,7 @@ docker run -u root --name mineru_docker \
| <vlm/hybrid>-auto-engine |
- 🟡 |
+ 🟢 |
| <vlm/hybrid>-http-client |
@@ -98,7 +99,7 @@ docker run -u root --name mineru_docker \
| <vlm/hybrid>-auto-engine |
- 🟡 |
+ 🟢 |
| <vlm/hybrid>-http-client |
diff --git a/docs/zh/usage/acceleration_cards/THead.md b/docs/zh/usage/acceleration_cards/THead.md
index ac916095..94ea525e 100644
--- a/docs/zh/usage/acceleration_cards/THead.md
+++ b/docs/zh/usage/acceleration_cards/THead.md
@@ -127,7 +127,7 @@ docker run --privileged=true \
| 数据并行 (--data-parallel-size/--dp) |
- 🟡 |
+ 🔴 |
🔴 |
diff --git a/docs/zh/usage/acceleration_cards/Tecorigin.md b/docs/zh/usage/acceleration_cards/Tecorigin.md
index 51765fb4..d490de4b 100644
--- a/docs/zh/usage/acceleration_cards/Tecorigin.md
+++ b/docs/zh/usage/acceleration_cards/Tecorigin.md
@@ -104,7 +104,7 @@ docker run -dit --name mineru_docker \
| 数据并行 (--data-parallel-size) |
- 🟡 |
+ 🔴 |
diff --git a/mineru/backend/vlm/utils.py b/mineru/backend/vlm/utils.py
index 9ea7ab21..07540985 100644
--- a/mineru/backend/vlm/utils.py
+++ b/mineru/backend/vlm/utils.py
@@ -189,17 +189,16 @@ def _apply_server_config(args: list, config: dict) -> None:
"""应用 server 模式的配置"""
import json
- if "compilation_config_dict" in config:
- _add_server_arg_if_missing(
- args, "compilation-config",
- json.dumps(config["compilation_config_dict"], separators=(',', ':'))
- )
-
- for key in ["block_size", "dtype", "distributed_executor_backend"]:
- if key in config:
+ for key, value in config.items():
+ if key == "compilation_config_dict":
+ _add_server_arg_if_missing(
+ args, "compilation-config",
+ json.dumps(value, separators=(',', ':'))
+ )
+ else:
# 转换 key 格式: block_size -> block-size
arg_name = key.replace("_", "-")
- _add_server_arg_if_missing(args, arg_name, str(config[key]))
+ _add_server_arg_if_missing(args, arg_name, str(value))
def _apply_engine_config(kwargs: dict, config: dict, vllm_mode: str) -> None:
@@ -209,16 +208,14 @@ def _apply_engine_config(kwargs: dict, config: dict, vllm_mode: str) -> None:
except ImportError:
raise ImportError("Please install vllm to use the vllm-async-engine backend.")
- if "compilation_config_dict" in config:
- config_dict = config["compilation_config_dict"]
- if vllm_mode == "sync_engine":
- compilation_config = config_dict
- elif vllm_mode == "async_engine":
- compilation_config = CompilationConfig(**config_dict)
+ for key, value in config.items():
+ if key == "compilation_config_dict":
+ if vllm_mode == "sync_engine":
+ compilation_config = value
+ elif vllm_mode == "async_engine":
+ compilation_config = CompilationConfig(**value)
+ else:
+ continue
+ _add_engine_kwarg_if_missing(kwargs, "compilation_config", compilation_config)
else:
- return
- _add_engine_kwarg_if_missing(kwargs, "compilation_config", compilation_config)
-
- for key in ["block_size", "dtype", "distributed_executor_backend"]:
- if key in config:
- _add_engine_kwarg_if_missing(kwargs, key, config[key])
+ _add_engine_kwarg_if_missing(kwargs, key, value)