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

Merge pull request #2644 from myhloli/dev

Browse Source 
fix: update configure_model to use environment variable for config file name
This commit is contained in:
Xiaomeng Zhao
2025-06-14 22:22:20 +08:00
committed by GitHub
parent 6e54a68cef a812ae899e
commit d77d5edddb
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
mineru/cli/models_download.py
View File

@@ -43,7 +43,7 @@ def download_and_modify_json(url, local_filename, modifications):
def configure_model(model_dir, model_type):
"""配置模型"""
json_url = 'https://gcore.jsdelivr.net/gh/opendatalab/MinerU@master/mineru.template.json'
config_file_name = 'mineru.json'
config_file_name = os.getenv('MINERU_TOOLS_CONFIG_JSON', 'mineru.json')
home_dir = os.path.expanduser('~')
config_file = os.path.join(home_dir, config_file_name)
@@ -119,13 +119,13 @@ def download_models(model_source, model_type):
click.echo(f"Downloading model: {model_path}")
download_finish_path = auto_download_and_get_model_root_path(model_path, repo_mode='pipeline')
click.echo(f"Pipeline models downloaded successfully to: {download_finish_path}")
configure_model(download_finish_path, model_type)
configure_model(download_finish_path, "pipeline")
def download_vlm_models():
"""下载VLM模型"""
download_finish_path = auto_download_and_get_model_root_path("/", repo_mode='vlm')
click.echo(f"VLM models downloaded successfully to: {download_finish_path}")
configure_model(download_finish_path, model_type)
configure_model(download_finish_path, "vlm")
try:
if model_type == 'pipeline':