|
@@ -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:
|
|
if os.getenv('MINERU_VIRTUAL_VRAM_SIZE', None) is None:
|
|
|
os.environ['MINERU_VIRTUAL_VRAM_SIZE']= str(get_virtual_vram_size())
|
|
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.environ['MINERU_MODEL_SOURCE'] = model_source
|
|
|
|
|
|
|
|
os.makedirs(output_dir, exist_ok=True)
|
|
os.makedirs(output_dir, exist_ok=True)
|