model_source.md 2.5 KB
模型源说明
MinerU使用 HuggingFace 和 ModelScope 作为模型仓库，用户可以根据需要切换模型源或使用本地模型。
HuggingFace是默认的模型源，在全球范围内提供了优异的加载速度和极高稳定性。ModelScope是中国大陆地区用户的最佳选择，提供了无缝兼容的SDK模块，适用于无法访问HuggingFace的用户。
模型源的切换方法
通过命令行参数切换
目前仅mineru命令行工具支持通过命令行参数切换模型源，其他命令行工具如mineru-api、mineru-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>