Browse Source

Update mineru/backend/vlm/custom_logits_processors.py

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
Xiaomeng Zhao 1 tháng trước cách đây
mục cha
commit
7c55a0ea65
1 tập tin đã thay đổi với 5 bổ sung4 xóa
  1. 5 4
      mineru/backend/vlm/custom_logits_processors.py

+ 5 - 4
mineru/backend/vlm/custom_logits_processors.py

@@ -18,11 +18,12 @@ def enable_custom_logits_processors():
     compute_capability = float(compute_capability)
 
     # 安全地处理环境变量
-    try:
-        vllm_use_v1 = int(os.getenv('VLLM_USE_V1', "1"))
-    except (ValueError, TypeError):
+    vllm_use_v1_str = os.getenv('VLLM_USE_V1', "1")
+    if vllm_use_v1_str.isdigit():
+        vllm_use_v1 = int(vllm_use_v1_str)
+    else:
         vllm_use_v1 = 1
-        logger.warning("Invalid VLLM_USE_V1 value")
+        logger.warning(f"Invalid VLLM_USE_V1 value: {vllm_use_v1_str!r}, defaulting to 1")
 
     if vllm_use_v1 == 0:
         logger.info("VLLM_USE_V1 is set to 0, disabling custom_logits_processors")