x

docker/scripts/patch_vllm_platform.py

@@ -95,6 +95,20 @@ def ensure_vllm_dist_info():
         return
 
     version = getattr(vllm, '__version__', '0.1.dev1') or '0.1.dev1'
+    # PEP440 合法性校验：vllm 在 _version.py 缺失时会回退到 'dev'，
+    # 这不是合法 PEP440，会触发下游 packaging.version.parse 抛 InvalidVersion
+    # （例如 mineru.backend.vlm.utils:set_default_gpu_memory_utilization）。
+    try:
+        from packaging.version import Version as _PEP440Version
+        _PEP440Version(version)
+    except Exception:
+        print(f'WARNING: vllm.__version__={version!r} is not PEP440-compliant; '
+              f'falling back to "0.11.0" for dist-info.')
+        version = '0.11.0'
+        try:
+            vllm.__version__ = version
+        except Exception:
+            pass
     site_packages = sysconfig.get_paths().get('purelib')
     if not site_packages:
         return