Browse Source

Update mineru/cli/vlm_server.py

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
Xiaomeng Zhao 4 days ago
parent
commit
096717e4d0
1 changed files with 2 additions and 1 deletions
  1. 2 1
      mineru/cli/vlm_server.py

+ 2 - 1
mineru/cli/vlm_server.py

@@ -32,10 +32,11 @@ def openai_server(ctx, inference_engine):
             inference_engine = 'vllm'
             logger.info("Using vLLM as the inference engine for VLM server.")
         except ImportError:
-            logger.info("vLLM not found, falling back to LMDeploy as the inference engine for VLM server.")
+            logger.info("vLLM not found, attempting to use LMDeploy as the inference engine for VLM server.")
             try:
                 import lmdeploy
                 inference_engine = 'lmdeploy'
+            # Success message moved after successful import
                 logger.info("Using LMDeploy as the inference engine for VLM server.")
             except ImportError:
                 logger.error("Neither vLLM nor LMDeploy is installed. Please install at least one of them.")