Browse Source

refactor: rename environment variable from MINERU_BACKEND to MINERU_MODEL_SOURCE

myhloli 5 months ago
parent
commit
8155480433
1 changed files with 1 additions and 1 deletions
  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)