Jelajahi Sumber

fix: add timing log for model predictor retrieval in vlm_analyze.py

myhloli 2 bulan lalu
induk
melakukan
c4eae4e0ef
1 mengubah file dengan 3 tambahan dan 0 penghapusan
  1. 3 0
      mineru/backend/vlm/vlm_analyze.py

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

@@ -31,6 +31,7 @@ class ModelSingleton:
     ) -> MinerUClient:
         key = (backend, model_path, server_url)
         if key not in self._models:
+            start_time = time.time()
             model = None
             processor = None
             vllm_llm = None
@@ -84,6 +85,8 @@ class ModelSingleton:
                 vllm_llm=vllm_llm,
                 server_url=server_url,
             )
+            elapsed = round(time.time() - start_time, 2)
+            logger.info(f"get {backend} predictor cost: {elapsed}s")
         return self._models[key]