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

feat: ensure compilation_config is a valid JSON string in vlm_analyze

Browse Source
This commit is contained in:
myhloli
2026-01-22 19:40:44 +08:00
parent fb046d3a5a
commit 04aceee288
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
mineru/backend/vlm/vlm_analyze.py
View File

@@ -144,13 +144,17 @@ class ModelSingleton:
"simple_cuda_graph": True
}
block_size = 32
kwargs["compilation_config"] = compilation_config
kwargs["compilation_config"] = json.dumps(compilation_config)
kwargs["block_size"] = block_size
if "compilation_config" in kwargs:
if isinstance(kwargs["compilation_config"], str):
if isinstance(kwargs["compilation_config"], dict):
# 如果是字典，转换为 JSON 字符串
kwargs["compilation_config"] = json.dumps(kwargs["compilation_config"])
elif isinstance(kwargs["compilation_config"], str):
# 验证是否为有效 JSON
try:
kwargs["compilation_config"] = json.loads(kwargs["compilation_config"])
json.loads(kwargs["compilation_config"])
except json.JSONDecodeError:
logger.warning(
f"Failed to parse compilation_config as JSON: {kwargs['compilation_config']}")