Преглед на файлове

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