Unverified Commit d77d5edd authored by Xiaomeng Zhao's avatar Xiaomeng Zhao Committed by GitHub
Browse files

Merge pull request #2644 from myhloli/dev

fix: update configure_model to use environment variable for config file name
parents 6e54a68c a812ae89
...@@ -43,7 +43,7 @@ def download_and_modify_json(url, local_filename, modifications): ...@@ -43,7 +43,7 @@ def download_and_modify_json(url, local_filename, modifications):
def configure_model(model_dir, model_type): def configure_model(model_dir, model_type):
"""配置模型""" """配置模型"""
json_url = 'https://gcore.jsdelivr.net/gh/opendatalab/MinerU@master/mineru.template.json' 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('~') home_dir = os.path.expanduser('~')
config_file = os.path.join(home_dir, config_file_name) config_file = os.path.join(home_dir, config_file_name)
...@@ -119,13 +119,13 @@ def download_models(model_source, model_type): ...@@ -119,13 +119,13 @@ def download_models(model_source, model_type):
click.echo(f"Downloading model: {model_path}") click.echo(f"Downloading model: {model_path}")
download_finish_path = auto_download_and_get_model_root_path(model_path, repo_mode='pipeline') 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}") 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(): def download_vlm_models():
"""下载VLM模型""" """下载VLM模型"""
download_finish_path = auto_download_and_get_model_root_path("/", repo_mode='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}") 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: try:
if model_type == 'pipeline': if model_type == 'pipeline':
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment