Create a link to DotsOCR

patch_vllm.sh

@@ -7,9 +7,9 @@ set -e
 echo "Patching vLLM entrypoint to register DotsOCR model..."
 
 # Patch the vLLM script to import the DotsOCR modeling code
-# NOTE: Import directly from the model folder on PYTHONPATH, not from a non-existent 'DotsOCR' package
+# Import via the DotsOCR package so relative imports inside the module resolve
 sed -i '/^from vllm\.entrypoints\.cli\.main import main$/a\
-import modeling_dots_ocr_vllm' $(which vllm)
+from DotsOCR import modeling_dots_ocr_vllm' $(which vllm)
 
 echo "vLLM script after patch:"
 grep -A 1 'from vllm.entrypoints.cli.main import main' $(which vllm)

start_server.sh

@@ -11,11 +11,15 @@ fi
 # Ensure Python sees the model's modules (e.g., modeling_dots_ocr_vllm.py).
 export PYTHONPATH="${HF_MODEL_PATH}:${PYTHONPATH}"
 
+if [ -n "$HF_MODEL_PATH" ] && [ ! -e "DotsOCR" ]; then
+    ln -s "$HF_MODEL_PATH" DotsOCR
+fi
+
 python3 - <<'PY'
 import sys, os
 print("HF_MODEL_PATH:", os.environ.get("HF_MODEL_PATH"))
 print("PYTHONPATH:", sys.path)
-import modeling_dots_ocr_vllm
+from DotsOCR import modeling_dots_ocr_vllm
 print("DotsOCR import OK")
 PY