Pārlūkot izejas kodu

fix: import enable_custom_logits_processors in server.py

myhloli 3 nedēļas atpakaļ
vecāks
revīzija
7c77db7135
1 mainītis faili ar 1 papildinājumiem un 3 dzēšanām
  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