centersl/MinerU
1
0
Fork 0
mirror of https://github.com/opendatalab/MinerU.git synced 2026-03-27 02:58:54 +07:00
Code Issues Packages Projects Releases Wiki Activity

fix: update status indicators in documentation and improve config handling in utils.py

Browse Source
This commit is contained in:
myhloli
2026-02-05 15:09:24 +08:00
parent 6e7c6b082d
commit 97450688d6
4 changed files with 24 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
docs/zh/usage/acceleration_cards/Kunlunxin.md
View File

@@ -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 \
</tr>
<tr>
<td>&lt;vlm/hybrid&gt;-auto-engine</td>
<td>🟡</td>
<td>🟢</td>
</tr>
<tr>
<td>&lt;vlm/hybrid&gt;-http-client</td>
@@ -85,7 +86,7 @@ docker run -u root --name mineru_docker \
</tr>
<tr>
<td>&lt;vlm/hybrid&gt;-auto-engine</td>
<td>🟡</td>
<td>🟢</td>
</tr>
<tr>
<td>&lt;vlm/hybrid&gt;-http-client</td>
@@ -98,7 +99,7 @@ docker run -u root --name mineru_docker \
</tr>
<tr>
<td>&lt;vlm/hybrid&gt;-auto-engine</td>
<td>🟡</td>
<td>🟢</td>
</tr>
<tr>
<td>&lt;vlm/hybrid&gt;-http-client</td>

2
docs/zh/usage/acceleration_cards/THead.md
View File

@@ -127,7 +127,7 @@ docker run --privileged=true \
</tr>
<tr>
<td colspan="2">数据并行 (--data-parallel-size/--dp)</td>
<td>🟡</td>
<td>🔴</td>
<td>🔴</td>
</tr>
</tbody>

2
docs/zh/usage/acceleration_cards/Tecorigin.md
View File

@@ -104,7 +104,7 @@ docker run -dit --name mineru_docker \
</tr>
<tr>
<td colspan="2">数据并行 (--data-parallel-size)</td>
<td>🟡</td>
<td>🔴</td>
</tr>
</tbody>
</table>

39
mineru/backend/vlm/utils.py
View File

@@ -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)