瀏覽代碼

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