Browse Source

refactor: add 'local' option for model source and comment out timing logs

myhloli 5 months ago
parent
commit
ab93f856f5
2 changed files with 4 additions and 4 deletions
  1. 3 3
      mineru/backend/vlm/vlm_analyze.py
  2. 1 1
      mineru/cli/client.py

+ 3 - 3
mineru/backend/vlm/vlm_analyze.py

@@ -47,11 +47,11 @@ def doc_analyze(
     if predictor is None:
         predictor = ModelSingleton().get_model(backend, model_path, server_url)
 
-    load_images_start = time.time()
+    # load_images_start = time.time()
     images_list, pdf_doc = load_images_from_pdf(pdf_bytes)
     images_base64_list = [image_dict["img_base64"] for image_dict in images_list]
-    load_images_time = round(time.time() - load_images_start, 2)
-    logger.info(f"load images cost: {load_images_time}, speed: {round(len(images_base64_list)/load_images_time, 3)} images/s")
+    # load_images_time = round(time.time() - load_images_start, 2)
+    # logger.info(f"load images cost: {load_images_time}, speed: {round(len(images_base64_list)/load_images_time, 3)} images/s")
 
     infer_start = time.time()
     results = predictor.batch_predict(images=images_base64_list)

+ 1 - 1
mineru/cli/client.py

@@ -129,7 +129,7 @@ from .common import do_parse, read_fn, pdf_suffixes, image_suffixes
     '-r',
     '--repo',
     'model_source',
-    type=click.Choice(['huggingface', 'modelscope']),
+    type=click.Choice(['huggingface', 'modelscope', 'local']),
     help="""
     The source of the model repository. Default is 'huggingface'.
     """,