소스 검색

fix: import enable_custom_logits_processors in server.py

myhloli 3 주 전
부모
커밋
7c77db7135
1개의 변경된 파일1개의 추가작업 그리고 3개의 파일을 삭제
  1. 1 3
      mineru/model/vlm_vllm_model/server.py

+ 1 - 3
mineru/model/vlm_vllm_model/server.py

@@ -1,9 +1,7 @@
 import os
 import sys
 
-from mineru.backend.vlm.custom_logits_processors import enable_custom_logits_processors
-
-from mineru.backend.vlm.utils import set_defult_gpu_memory_utilization
+from mineru.backend.vlm.utils import set_defult_gpu_memory_utilization, enable_custom_logits_processors
 from mineru.utils.models_download_utils import auto_download_and_get_model_root_path
 
 from vllm.entrypoints.cli.main import main as vllm_main