|
|
@@ -27,13 +27,13 @@ DEFAULT_API_URL = _config.get_str("api_url", "http://127.0.0.1:5282")
|
|
|
|
|
|
# 根据设备类型自动选择 MinerU backend(配置文件显式指定时优先)
|
|
|
# npu -> vlm-http-client(NPU 不支持本地 vllm,需通过 http 调用外部推理服务)
|
|
|
-# nvi -> vlm-async-engine(NVIDIA GPU 本地异步推理)
|
|
|
+# nvi -> vlm-vllm-async-engine(NVIDIA GPU 本地异步推理,API会自动去掉vlm-前缀)
|
|
|
# cpu -> pipeline(纯 CPU 通用管线)
|
|
|
def _auto_backend() -> str:
|
|
|
if DEVICE_KIND == "npu":
|
|
|
return "vlm-http-client"
|
|
|
elif DEVICE_KIND == "nvi":
|
|
|
- return "vlm-async-engine"
|
|
|
+ return "vlm-vllm-async-engine"
|
|
|
return "pipeline"
|
|
|
|
|
|
def _auto_server_url() -> str:
|