Browse Source

Add support for --logits-processors argument in server.py

myhloli 2 months ago
parent
commit
bce9bb6d1d
1 changed files with 2 additions and 2 deletions
  1. 2 2
      mineru/model/vlm_vllm_model/server.py

+ 2 - 2
mineru/model/vlm_vllm_model/server.py

@@ -22,6 +22,8 @@ def main():
             has_port_arg = True
         if arg == "--gpu-memory-utilization" or arg.startswith("--gpu-memory-utilization="):
             has_gpu_memory_utilization_arg = True
+        if arg == "--logits-processors" or arg.startswith("--logits-processors="):
+            has_logits_processors_arg = True
         if arg == "--model":
             if i + 1 < len(args):
                 model_path = args[i + 1]
@@ -29,8 +31,6 @@ def main():
         elif arg.startswith("--model="):
             model_path = arg.split("=", 1)[1]
             model_arg_indices.append(i)
-        if arg == "--logits-processors" or arg.startswith("--logits-processors="):
-            has_logits_processors_arg = True
 
     # 从参数列表中移除 --model 参数
     if model_arg_indices: