浏览代码

refactor: rename environment variable from MINERU_BACKEND to MINERU_MODEL_SOURCE

myhloli 5 月之前
父节点
当前提交
8155480433
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      mineru/cli/client.py

+ 1 - 1
mineru/cli/client.py

@@ -151,7 +151,7 @@ def main(input_path, output_dir, backend, lang, server_url, start_page_id, end_p
     if os.getenv('MINERU_VIRTUAL_VRAM_SIZE', None) is None:
         os.environ['MINERU_VIRTUAL_VRAM_SIZE']= str(get_virtual_vram_size())
 
-    if os.getenv('MINERU_BACKEND', None) is None:
+    if os.getenv('MINERU_MODEL_SOURCE', None) is None:
         os.environ['MINERU_MODEL_SOURCE'] = model_source
 
     os.makedirs(output_dir, exist_ok=True)