跳转至

模型源说明

MinerU使用 HuggingFaceModelScope 作为模型仓库,用户可以根据需要切换模型源或使用本地模型。

  • HuggingFace 是默认的模型源,在全球范围内提供了优异的加载速度和极高稳定性。
  • ModelScope 是中国大陆地区用户的最佳选择,提供了无缝兼容的SDK模块,适用于无法访问HuggingFace的用户。

模型源的切换方法

通过命令行参数切换

目前仅mineru命令行工具支持通过命令行参数切换模型源,其他命令行工具如mineru-apimineru-gradio等暂不支持。

mineru -p <input_path> -o <output_path> --source modelscope

通过环境变量切换

在任何情况下可以通过设置环境变量来切换模型源,这适用于所有命令行工具和API调用。

export MINERU_MODEL_SOURCE=modelscope
import os
os.environ["MINERU_MODEL_SOURCE"] = "modelscope"

Tip

通过环境变量设置的模型源会在当前终端会话中生效,直到终端关闭或环境变量被修改。且优先级高于命令行参数,如同时设置了命令行参数和环境变量,命令行参数将被忽略。

使用本地模型

1. 下载模型到本地

mineru-models-download --help
或使用交互式命令行工具选择模型下载:
mineru-models-download

Note

  • 下载完成后,模型路径会在当前终端窗口输出,并自动写入用户目录下的 mineru.json
  • 您也可以通过将配置模板文件复制到用户目录下并重命名为 mineru.json 来创建配置文件。
  • 模型下载到本地后,您可以自由移动模型文件夹到其他位置,同时需要在 mineru.json 中更新模型路径。
  • 如您将模型文件夹部署到其他服务器上,请确保将 mineru.json文件一同移动到新设备的用户目录中并正确配置模型路径。
  • 如您需要更新模型文件,可以再次运行 mineru-models-download 命令,模型更新暂不支持自定义路径,如您没有移动本地模型文件夹,模型文件会增量更新;如您移动了模型文件夹,模型文件会重新下载到默认位置并更新mineru.json

2. 使用本地模型进行解析

mineru -p <input_path> -o <output_path> --source local
或通过环境变量启用:
export MINERU_MODEL_SOURCE=local
mineru -p <input_path> -o <output_path>